From ffb127e0fbd912efa73f242ea1b7caca0101e5b1 Mon Sep 17 00:00:00 2001 From: Taylor Mordan <48797707+taylormordan@users.noreply.github.com> Date: Thu, 7 Mar 2024 17:46:26 +0100 Subject: [PATCH] Add checkpoints (#22) * add checkpoints for 5 backbones * remove ssl for checkpoints --- src/openpifpaf/plugins/coco/__init__.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/src/openpifpaf/plugins/coco/__init__.py b/src/openpifpaf/plugins/coco/__init__.py index 13ca790f9..57232971b 100644 --- a/src/openpifpaf/plugins/coco/__init__.py +++ b/src/openpifpaf/plugins/coco/__init__.py @@ -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/' @@ -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'] = (