Skip to content

Commit

Permalink
update weights url
Browse files Browse the repository at this point in the history
  • Loading branch information
gau-nernst committed Jul 7, 2023
1 parent 1f3c244 commit d12d1b5
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 9 deletions.
2 changes: 1 addition & 1 deletion vision_toolbox/backbones/darknet.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def get_feature_maps(self, x: Tensor) -> list[Tensor]:
**_base,
num_blocks_list=(1, 2, 8, 8, 4),
stage_cls=CSPDarknetStage,
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/cspdarknet53-f8225d3d.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/cspdarknet53-3bfa0423.pth",
),
# from YOLOv5
"darknet-yolov5n": dict(
Expand Down
16 changes: 8 additions & 8 deletions vision_toolbox/backbones/vovnet.py
Original file line number Diff line number Diff line change
Expand Up @@ -39,48 +39,48 @@
**_slim,
"num_blocks_list": (1, 1, 1, 1),
"ese": False,
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet27_slim-79617b9c.pth",
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet27_slim-dd43306a.pth",
},
"vovnet-39": dict(
**_base,
num_blocks_list=(1, 1, 2, 2),
ese=False,
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet39-ced4435d.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet39-4c79d629.pth",
),
"vovnet-57": dict(
**_base,
num_blocks_list=(1, 1, 4, 3),
ese=False,
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet57-9929476e.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet57-ecb9cc34.pth",
),
# VoVNetV2
"vovnet-19-slim-ese": {
**_base,
**_slim,
"num_layers_list": (3, 3, 3, 3),
"num_blocks_list": (1, 1, 1, 1),
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet19_slim_ese-446e2ae9.pth",
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet19_slim_ese-f8075640.pth",
},
"vovnet-19-ese": {
**_base,
"num_layers_list": (3, 3, 3, 3),
"num_blocks_list": (1, 1, 1, 1),
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet19_ese-4410fc5f.pth",
"weights": "https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet19_ese-a077657e.pth",
},
"vovnet-39-ese": dict(
**_base,
num_blocks_list=(1, 1, 2, 2),
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet39_ese-b73bdbe9.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet39_ese-9ce81b0d.pth",
),
"vovnet-57-ese": dict(
**_base,
num_blocks_list=(1, 1, 4, 3),
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet57_ese-630a88d1.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet57_ese-ae1a7f89.pth",
),
"vovnet-99-ese": dict(
**_base,
num_blocks_list=(1, 3, 9, 3),
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet99_ese-56fd52f5.pth",
weights="https://github.com/gau-nernst/vision-toolbox/releases/download/v0.0.1/vovnet99_ese-713f3062.pth",
),
}

Expand Down

0 comments on commit d12d1b5

Please sign in to comment.