File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -220,7 +220,7 @@ way of doing it:
220
220
import torchvision
221
221
from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
222
222
223
- # load a model pre-trained pre-trained on COCO
223
+ # load a model pre-trained on COCO
224
224
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained = True )
225
225
226
226
# replace the classifier with a new one, that has
@@ -290,7 +290,7 @@ be using Mask R-CNN:
290
290
291
291
292
292
def get_model_instance_segmentation (num_classes ):
293
- # load an instance segmentation model pre-trained pre-trained on COCO
293
+ # load an instance segmentation model pre-trained on COCO
294
294
model = torchvision.models.detection.maskrcnn_resnet50_fpn(pretrained = True )
295
295
296
296
# get number of input features for the classifier
You can’t perform that action at this time.
0 commit comments