@@ -372,7 +372,7 @@ def fasterrcnn_resnet50_fpn(pretrained=False, progress=True,
372
372
373
373
374
374
def fasterrcnn_mobilenet_v3_large (pretrained = False , progress = True , num_classes = 91 , pretrained_backbone = True ,
375
- trainable_backbone_layers = None , min_size = 320 , max_size = 640 , ** kwargs ):
375
+ trainable_backbone_layers = None , min_size = 320 , max_size = 320 , ** kwargs ):
376
376
"""
377
377
Constructs a Faster R-CNN model with a MobileNetV3-Large backbone. It works similarly
378
378
to Faster R-CNN with ResNet-50 FPN backbone. See `fasterrcnn_resnet50_fpn` for more details.
@@ -402,7 +402,7 @@ def fasterrcnn_mobilenet_v3_large(pretrained=False, progress=True, num_classes=9
402
402
backbone = mobilenet_backbone ("mobilenet_v3_large" , pretrained_backbone , False ,
403
403
trainable_layers = trainable_backbone_layers )
404
404
405
- anchor_sizes = ((16 , 32 , 64 , 128 , 256 ), )
405
+ anchor_sizes = ((8 , 16 , 32 , 64 , 128 , ), )
406
406
aspect_ratios = ((0.5 , 1.0 , 2.0 ), )
407
407
408
408
model = FasterRCNN (backbone , num_classes , rpn_anchor_generator = AnchorGenerator (anchor_sizes , aspect_ratios ),
@@ -414,7 +414,7 @@ def fasterrcnn_mobilenet_v3_large(pretrained=False, progress=True, num_classes=9
414
414
415
415
416
416
def fasterrcnn_mobilenet_v3_large_fpn (pretrained = False , progress = True , num_classes = 91 , pretrained_backbone = True ,
417
- trainable_backbone_layers = None , min_size = 320 , max_size = 640 , ** kwargs ):
417
+ trainable_backbone_layers = None , min_size = 320 , max_size = 320 , ** kwargs ):
418
418
"""
419
419
Constructs a Faster R-CNN model with a MobileNetV3-Large FPN backbone. It works similarly
420
420
to Faster R-CNN with ResNet-50 FPN backbone. See `fasterrcnn_resnet50_fpn` for more details.
@@ -444,7 +444,7 @@ def fasterrcnn_mobilenet_v3_large_fpn(pretrained=False, progress=True, num_class
444
444
backbone = mobilenet_backbone ("mobilenet_v3_large" , pretrained_backbone , True ,
445
445
trainable_layers = trainable_backbone_layers )
446
446
447
- anchor_sizes = ((16 , 32 , 64 , 128 , 256 , ), ) * 3
447
+ anchor_sizes = ((8 , 16 , 32 , 64 , 128 , ), ) * 3
448
448
aspect_ratios = ((0.5 , 1.0 , 2.0 ),) * len (anchor_sizes )
449
449
450
450
model = FasterRCNN (backbone , num_classes , rpn_anchor_generator = AnchorGenerator (anchor_sizes , aspect_ratios ),
0 commit comments