Skip to content

Y
yolov7-obb

小毕学习代码 / yolov7-obb
与 Fork 源项目一致

从无法访问的项目Fork

通知 1
Star 1
Fork 0
Y
yolov7-obb

体验新版 GitCode，发现更多精彩内容 >>

提交 15d5cdad 编写于 作者: Bubbliiiing's avatar Bubbliiiing
浏览文件
操作

update a lot

上级 cdc66baa
隐藏空白更改
内联 并排
Showing with 4 addition and 4 deletion
nets/backbone.py
浏览文件 @ 15d5cdad
......@@ -110,8 +110,8 @@ class Backbone(nn.Module):
if pretrained:
url = {
"l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_backbone.pth',
"x" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_x_backbone.pth',
"l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_backbone_weights.pth',
"x" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_x_backbone_weights.pth',
}[phi]
checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", model_dir="./model_data")
self.load_state_dict(checkpoint, strict=False)
......
utils/utils.py
浏览文件 @ 15d5cdad
......@@ -76,8 +76,8 @@ def download_weights(phi, model_dir="./model_data"):
from torch.hub import load_state_dict_from_url
download_urls = {
"l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_backbone.pth',
"x" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_x_backbone.pth',
"l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_backbone_weights.pth',
"x" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/yolov7_x_backbone_weights.pth',
}
url = download_urls[phi]
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或