Skip to content

Commit bd0e7d8

Browse files
authored
add 3 facenet sub-models (#227)
1 parent 560af73 commit bd0e7d8

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tools/convert-models.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,9 @@ def blob_exist(bucket_name, blob_name):
7878
'deeplabv3_resnet101_coco': 'https://download.pytorch.org/models/deeplabv3_resnet101_coco-586e9e4e.pth',
7979
'casia-webface': 'https://github.com/timesler/facenet-pytorch/releases/download/v2.2.9/20180408-102900-casia-webface.pt',
8080
'vggface2': 'https://github.com/timesler/facenet-pytorch/releases/download/v2.2.9/20180402-114759-vggface2.pt',
81+
'facenet_onet': 'https://raw.githubusercontent.com/timesler/facenet-pytorch/refs/heads/master/data/onet.pt',
82+
'facenet_pnet': 'https://raw.githubusercontent.com/timesler/facenet-pytorch/refs/heads/master/data/pnet.pt',
83+
'facenet_rnet': 'https://raw.githubusercontent.com/timesler/facenet-pytorch/refs/heads/master/data/rnet.pt',
8184
'efficientnet_b0': 'https://download.pytorch.org/models/efficientnet_b0_rwightman-3dd342df.pth',
8285
'efficientnet_b1': 'https://download.pytorch.org/models/efficientnet_b1-c27df63c.pth', # IMAGENET1K_V2
8386
'efficientnet_b2': 'https://download.pytorch.org/models/efficientnet_b2_rwightman-bcdf34b7.pth',

0 commit comments

Comments
 (0)