Phân tích mã nguồn MMDetection: SingleRoIExtractor và BaseRoIExtractor

Lớp SingleRoIExtractor, được định nghĩa trong tệp single_level_roi_extractor.py, có nhiệm vụ trích xuất đặc trưng từ các vùng quan tâm (RoI) từ một hoặc nhiều lớp đặc trưng. Nó kế thừa từ lớp cơ sở BaseRoIExtractor. import torch from mmcv.runner import force_fp32 from mmdet.models.builder import ROI_EXTRACTORS from .base_roi_extractor import B ...

Đăng vào ngày 6 tháng 6 lúc 03:55

Phân tích mã nguồn MMDetection: Cấu trúc RPN Head trong Faster R-CNN

Trong cấu hình mô hình Faster R-CNN sử dụng ResNet-50 và FPN, phần rpn_head đóng vai trò then chốt trong việc đề xuất vùng ứng viên (region proposals). Dưới đây là phân tích chi tiết về lớp RPNHead — thành phần chịu trách nhiệm sinh anchor, dự đoán độ tin cậy và điều chỉnh bounding box. Cấu hình RPN Head rpn_head=dict( type='RPNHead', ...

Đăng vào ngày 22 tháng 5 lúc 17:50