关注RPN一栏目：
import torch
import torchvision
from torchvision.models.detection import FasterRCNN
from torchvision.models.detection.rpn import AnchorGenerator
# 定义backbone
backbone = torchvision.models.mobilenet_v2().features
backbone.FasterRCNN示例代码测试1：令anchor_generator = None
猜你喜欢
转载自blog.csdn.net/qq_45583898/article/details/125927213
今日推荐
周排行