1. Học hỏi
    2. /
  2. Khoa Học
    3. /
  3. Deep Learning cho Ảnh với PyTorch
Connected

Bài tập

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.

Hướng dẫn

100 XP
  • Import AnchorGenerator từ 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 AnchorGenerator với anchor_sizes và aspect_ratios.