Skip to content

Commit 4f03f8f

Browse files
zhiqwangzhiqwang
committed
Replace Coco to COCO in enum fields
1 parent fe65d37 commit 4f03f8f

File tree

6 files changed

+28
-28
lines changed

6 files changed

+28
-28
lines changed

‎torchvision/prototype/models/detection/faster_rcnn.py‎

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040

4141

4242
class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):
43-
Coco_V1 = Weights(
43+
COCO_V1 = Weights(
4444
url="https://download.pytorch.org/models/fasterrcnn_resnet50_fpn_coco-258fb6c6.pth",
4545
transforms=CocoEval,
4646
meta={
@@ -50,11 +50,11 @@ class FasterRCNN_ResNet50_FPN_Weights(WeightsEnum):
5050
"map": 37.0,
5151
},
5252
)
53-
default = Coco_V1
53+
default = COCO_V1
5454

5555

5656
class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):
57-
Coco_V1 = Weights(
57+
COCO_V1 = Weights(
5858
url="https://download.pytorch.org/models/fasterrcnn_mobilenet_v3_large_fpn-fb6a3cc7.pth",
5959
transforms=CocoEval,
6060
meta={
@@ -64,11 +64,11 @@ class FasterRCNN_MobileNet_V3_Large_FPN_Weights(WeightsEnum):
6464
"map": 32.8,
6565
},
6666
)
67-
default = Coco_V1
67+
default = COCO_V1
6868

6969

7070
class FasterRCNN_MobileNet_V3_Large_320_FPN_Weights(WeightsEnum):
71-
Coco_V1 = Weights(
71+
COCO_V1 = Weights(
7272
url="https://download.pytorch.org/models/fasterrcnn_mobilenet_v3_large_320_fpn-907ea3f9.pth",
7373
transforms=CocoEval,
7474
meta={
@@ -78,11 +78,11 @@ class FasterRCNN_MobileNet_V3_Large_320_FPN_Weights(WeightsEnum):
7878
"map": 22.8,
7979
},
8080
)
81-
default = Coco_V1
81+
default = COCO_V1
8282

8383

8484
@handle_legacy_interface(
85-
weights=("pretrained", FasterRCNN_ResNet50_FPN_Weights.Coco_V1),
85+
weights=("pretrained", FasterRCNN_ResNet50_FPN_Weights.COCO_V1),
8686
weights_backbone=("pretrained_backbone", ResNet50_Weights.ImageNet1K_V1),
8787
)
8888
def fasterrcnn_resnet50_fpn(
@@ -113,7 +113,7 @@ def fasterrcnn_resnet50_fpn(
113113

114114
if weights is not None:
115115
model.load_state_dict(weights.get_state_dict(progress=progress))
116-
if weights == FasterRCNN_ResNet50_FPN_Weights.Coco_V1:
116+
if weights == FasterRCNN_ResNet50_FPN_Weights.COCO_V1:
117117
overwrite_eps(model, 0.0)
118118

119119
return model
@@ -161,7 +161,7 @@ def _fasterrcnn_mobilenet_v3_large_fpn(
161161

162162

163163
@handle_legacy_interface(
164-
weights=("pretrained", FasterRCNN_MobileNet_V3_Large_FPN_Weights.Coco_V1),
164+
weights=("pretrained", FasterRCNN_MobileNet_V3_Large_FPN_Weights.COCO_V1),
165165
weights_backbone=("pretrained_backbone", MobileNet_V3_Large_Weights.ImageNet1K_V1),
166166
)
167167
def fasterrcnn_mobilenet_v3_large_fpn(
@@ -192,7 +192,7 @@ def fasterrcnn_mobilenet_v3_large_fpn(
192192

193193

194194
@handle_legacy_interface(
195-
weights=("pretrained", FasterRCNN_MobileNet_V3_Large_320_FPN_Weights.Coco_V1),
195+
weights=("pretrained", FasterRCNN_MobileNet_V3_Large_320_FPN_Weights.COCO_V1),
196196
weights_backbone=("pretrained_backbone", MobileNet_V3_Large_Weights.ImageNet1K_V1),
197197
)
198198
def fasterrcnn_mobilenet_v3_large_320_fpn(

‎torchvision/prototype/models/detection/keypoint_rcnn.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ class KeypointRCNN_ResNet50_FPN_Weights(WeightsEnum):
4545
"map_kp": 61.1,
4646
},
4747
)
48-
Coco_V1 = Weights(
48+
COCO_V1 = Weights(
4949
url="https://download.pytorch.org/models/keypointrcnn_resnet50_fpn_coco-fc266e95.pth",
5050
transforms=CocoEval,
5151
meta={
@@ -56,15 +56,15 @@ class KeypointRCNN_ResNet50_FPN_Weights(WeightsEnum):
5656
"map_kp": 65.0,
5757
},
5858
)
59-
default = Coco_V1
59+
default = COCO_V1
6060

6161

6262
@handle_legacy_interface(
6363
weights=(
6464
"pretrained",
6565
lambda kwargs: KeypointRCNN_ResNet50_FPN_Weights.Coco_Legacy
6666
if kwargs["pretrained"] == "legacy"
67-
else KeypointRCNN_ResNet50_FPN_Weights.Coco_V1,
67+
else KeypointRCNN_ResNet50_FPN_Weights.COCO_V1,
6868
),
6969
weights_backbone=("pretrained_backbone", ResNet50_Weights.ImageNet1K_V1),
7070
)
@@ -101,7 +101,7 @@ def keypointrcnn_resnet50_fpn(
101101

102102
if weights is not None:
103103
model.load_state_dict(weights.get_state_dict(progress=progress))
104-
if weights == KeypointRCNN_ResNet50_FPN_Weights.Coco_V1:
104+
if weights == KeypointRCNN_ResNet50_FPN_Weights.COCO_V1:
105105
overwrite_eps(model, 0.0)
106106

107107
return model

‎torchvision/prototype/models/detection/mask_rcnn.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424

2525

2626
class MaskRCNN_ResNet50_FPN_Weights(WeightsEnum):
27-
Coco_V1 = Weights(
27+
COCO_V1 = Weights(
2828
url="https://download.pytorch.org/models/maskrcnn_resnet50_fpn_coco-bf2d0c1e.pth",
2929
transforms=CocoEval,
3030
meta={
@@ -39,11 +39,11 @@ class MaskRCNN_ResNet50_FPN_Weights(WeightsEnum):
3939
"map_mask": 34.6,
4040
},
4141
)
42-
default = Coco_V1
42+
default = COCO_V1
4343

4444

4545
@handle_legacy_interface(
46-
weights=("pretrained", MaskRCNN_ResNet50_FPN_Weights.Coco_V1),
46+
weights=("pretrained", MaskRCNN_ResNet50_FPN_Weights.COCO_V1),
4747
weights_backbone=("pretrained_backbone", ResNet50_Weights.ImageNet1K_V1),
4848
)
4949
def maskrcnn_resnet50_fpn(
@@ -74,7 +74,7 @@ def maskrcnn_resnet50_fpn(
7474

7575
if weights is not None:
7676
model.load_state_dict(weights.get_state_dict(progress=progress))
77-
if weights == MaskRCNN_ResNet50_FPN_Weights.Coco_V1:
77+
if weights == MaskRCNN_ResNet50_FPN_Weights.COCO_V1:
7878
overwrite_eps(model, 0.0)
7979

8080
return model

‎torchvision/prototype/models/detection/retinanet.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626

2727
class RetinaNet_ResNet50_FPN_Weights(WeightsEnum):
28-
Coco_V1 = Weights(
28+
COCO_V1 = Weights(
2929
url="https://download.pytorch.org/models/retinanet_resnet50_fpn_coco-eeacb38b.pth",
3030
transforms=CocoEval,
3131
meta={
@@ -39,11 +39,11 @@ class RetinaNet_ResNet50_FPN_Weights(WeightsEnum):
3939
"map": 36.4,
4040
},
4141
)
42-
default = Coco_V1
42+
default = COCO_V1
4343

4444

4545
@handle_legacy_interface(
46-
weights=("pretrained", RetinaNet_ResNet50_FPN_Weights.Coco_V1),
46+
weights=("pretrained", RetinaNet_ResNet50_FPN_Weights.COCO_V1),
4747
weights_backbone=("pretrained_backbone", ResNet50_Weights.ImageNet1K_V1),
4848
)
4949
def retinanet_resnet50_fpn(
@@ -77,7 +77,7 @@ def retinanet_resnet50_fpn(
7777

7878
if weights is not None:
7979
model.load_state_dict(weights.get_state_dict(progress=progress))
80-
if weights == RetinaNet_ResNet50_FPN_Weights.Coco_V1:
80+
if weights == RetinaNet_ResNet50_FPN_Weights.COCO_V1:
8181
overwrite_eps(model, 0.0)
8282

8383
return model

‎torchvision/prototype/models/detection/ssd.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323

2424

2525
class SSD300_VGG16_Weights(WeightsEnum):
26-
Coco_V1 = Weights(
26+
COCO_V1 = Weights(
2727
url="https://download.pytorch.org/models/ssd300_vgg16_coco-b556d3b4.pth",
2828
transforms=CocoEval,
2929
meta={
@@ -38,11 +38,11 @@ class SSD300_VGG16_Weights(WeightsEnum):
3838
"map": 25.1,
3939
},
4040
)
41-
default = Coco_V1
41+
default = COCO_V1
4242

4343

4444
@handle_legacy_interface(
45-
weights=("pretrained", SSD300_VGG16_Weights.Coco_V1),
45+
weights=("pretrained", SSD300_VGG16_Weights.COCO_V1),
4646
weights_backbone=("pretrained_backbone", VGG16_Weights.ImageNet1K_Features),
4747
)
4848
def ssd300_vgg16(

‎torchvision/prototype/models/detection/ssdlite.py‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@
2828

2929

3030
class SSDLite320_MobileNet_V3_Large_Weights(WeightsEnum):
31-
Coco_V1 = Weights(
31+
COCO_V1 = Weights(
3232
url="https://download.pytorch.org/models/ssdlite320_mobilenet_v3_large_coco-a79551df.pth",
3333
transforms=CocoEval,
3434
meta={
@@ -43,11 +43,11 @@ class SSDLite320_MobileNet_V3_Large_Weights(WeightsEnum):
4343
"map": 21.3,
4444
},
4545
)
46-
default = Coco_V1
46+
default = COCO_V1
4747

4848

4949
@handle_legacy_interface(
50-
weights=("pretrained", SSDLite320_MobileNet_V3_Large_Weights.Coco_V1),
50+
weights=("pretrained", SSDLite320_MobileNet_V3_Large_Weights.COCO_V1),
5151
weights_backbone=("pretrained_backbone", MobileNet_V3_Large_Weights.ImageNet1K_V1),
5252
)
5353
def ssdlite320_mobilenet_v3_large(

0 commit comments

Comments
 (0)