zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

This commit is contained in:
Glenn Jocher 2020-07-16 17:18:47 -07:00
parent 5387d4747d 84a946615a
commit 87475c0345
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
hubconf.py
View File

@ -37,9 +37,11 @@ def create(name, pretrained, channels, classes):
state_dict = {k: v for k, v in state_dict.items() if model.state_dict()[k].shape == v.shape} # filter state_dict = {k: v for k, v in state_dict.items() if model.state_dict()[k].shape == v.shape} # filter
model.load_state_dict(state_dict, strict=False) # load model.load_state_dict(state_dict, strict=False) # load
return model return model
except Exception as e: except Exception as e:
help_url = 'https://github.com/ultralytics/yolov5/issues/36' help_url = 'https://github.com/ultralytics/yolov5/issues/36'
print('%s\nCache maybe be out of date. Delete cache and retry. See %s for help.' % (e, help_url)) s = 'Cache maybe be out of date, deleting cache and retrying may solve this. See %s for help.' % help_url
raise Exception(s) from e
def yolov5s(pretrained=False, channels=3, classes=80): def yolov5s(pretrained=False, channels=3, classes=80):

4
requirements.txt
View File

@ -1,13 +1,13 @@
# pip install -U -r requirements.txt # pip install -U -r requirements.txt
Cython Cython
numpy==1.17.3 numpy>=1.18.5
opencv-python opencv-python
torch>=1.5.1 torch>=1.5.1
matplotlib matplotlib
pillow pillow
tensorboard tensorboard
PyYAML>=5.3 PyYAML>=5.3
torchvision torchvision>=0.6
scipy scipy
tqdm tqdm
git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI