Bộ tạo anchor
Nhóm của bạn đang phát triển các mô hình phát hiện đối tượng dựa trên kiến trúc Faster R-CNN và sử dụng backbone đã được huấn luyện sẵn. Nhiệm vụ của bạn là tạo các anchor box để làm hộp giới hạn tham chiếu nhằm đề xuất các vùng đối tượng tiềm năng.
Bạn sẽ tạo 9 anchor chuẩn (3 kích thước hộp và 3 tỉ lệ khung hình).
torch đã được nhập sẵn cho bạn.
Bài tập này là một phần của khóa học
Deep Learning cho Ảnh với PyTorch
Hướng dẫn bài tập
- Import
AnchorGeneratortừtorchvision.models.detection.rpn. - Cấu hình kích thước anchor với 3 giá trị:
((32, 64, 128),). - Cấu hình tỉ lệ khung hình với 3 giá trị
`((0.5, 1.0, 2.0),). - Khởi tạo
AnchorGeneratorvớianchor_sizesvàaspect_ratios.
Bài tập tương tác thực hành trực tiếp
Hãy thử làm bài tập này bằng cách hoàn thành đoạn mã mẫu này.
# Import AnchorGenerator
____
# Configure anchor size
anchor_sizes = ____
# Configure aspect ratio
aspect_ratios = ____
# Instantiate AnchorGenerator
rpn_anchor_generator = ____