也就是说,假设我想创建一个速度更快的R模型,而不是对COCO进行预培训,并在ImageNet上进行预培训,然后得到主干网,然后我做以下操作:这与和所表示的主干网的设置是一致的。然而,当我通过模型传递一个图像时,如果我直接设置一个resnet50,结果与我将得到的结果不一致。, extract from the Faster R-CNN, als
我对PyTorch非常陌生,我正在尝试使用对象检测模型来进行传输学习,以便学习如何检测新的数据集。collate_fn)# load Faster RCNN pre-trained model
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="FasterRCNN_ResNet50_FPN_Weights.COCO_V1") # get the number of i
Create / load model
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)in_featuresone
model.roi_heads.box_predictor = FastRCNNPredictor(in