Skip to content

Commit

Permalink
Add checkpoints (#22)
Browse files Browse the repository at this point in the history
* add checkpoints for 5 backbones

* remove ssl for checkpoints
  • Loading branch information
taylormordan committed Mar 7, 2024
1 parent dd32c38 commit ffb127e
Showing 1 changed file with 15 additions and 0 deletions.
15 changes: 15 additions & 0 deletions src/openpifpaf/plugins/coco/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,9 @@ def register():
openpifpaf.CHECKPOINT_URLS['shufflenetv2k30'] = (
'http://github.com/openpifpaf/torchhub/releases/download/v0.13/'
'shufflenetv2k30-210821-003923-cocokp-slurm726072-edge513-o10s-5fe1c400.pkl')
openpifpaf.CHECKPOINT_URLS['shufflenetv2k30*'] = (
'http://github.com/vita-epfl/openpifpaf-torchhub/releases/download/v0.14.2/'
'cocokp-shufflenetv2k30.pt.epoch350')
openpifpaf.CHECKPOINT_URLS['shufflenetv2k44'] = openpifpaf.PRETRAINED_UNAVAILABLE
openpifpaf.CHECKPOINT_URLS['swin_s'] = (
'http://github.com/dmizr/openpifpaf/releases/download/'
Expand All @@ -46,6 +49,18 @@ def register():
openpifpaf.CHECKPOINT_URLS['swin_t_input_upsample'] = (
'http://github.com/dmizr/openpifpaf/releases/download/'
'v0.12.14/swin_t_input_upsample_no_fpn_lr_5e-5_resumed-e0681112.pkl')
openpifpaf.CHECKPOINT_URLS['swin_l_input_upsample'] = (
'http://github.com/vita-epfl/openpifpaf-torchhub/releases/download/v0.14.2/'
'cocokp-swinl-inputupsample.pt.epoch250')
openpifpaf.CHECKPOINT_URLS['hrformerbasecat'] = (
'http://github.com/vita-epfl/openpifpaf-torchhub/releases/download/v0.14.2/'
'cocokp-hrformerbasecat.pt.epoch250')
openpifpaf.CHECKPOINT_URLS['clipconvnextbase'] = (
'http://github.com/vita-epfl/openpifpaf-torchhub/releases/download/v0.14.2/'
'cocokp-clipconvnextbase.pt.epoch250')
openpifpaf.CHECKPOINT_URLS['convnextv2base'] = (
'http://github.com/vita-epfl/openpifpaf-torchhub/releases/download/v0.14.2/'
'cocokp-convnextv2base.pt.epoch250')

# object detection
openpifpaf.CHECKPOINT_URLS['mobilenetv3small-cocodet'] = (
Expand Down

0 comments on commit ffb127e

Please sign in to comment.