glenn-jocher commited on
Commit
87475c0
·
2 Parent(s): 5387d47 84a9466

Merge remote-tracking branch 'origin/master'

Browse files
Files changed (2) hide show
  1. hubconf.py +3 -1
  2. requirements.txt +2 -2
hubconf.py CHANGED
@@ -37,9 +37,11 @@ def create(name, pretrained, channels, classes):
37
  state_dict = {k: v for k, v in state_dict.items() if model.state_dict()[k].shape == v.shape} # filter
38
  model.load_state_dict(state_dict, strict=False) # load
39
  return model
 
40
  except Exception as e:
41
  help_url = 'https://github.com/ultralytics/yolov5/issues/36'
42
- print('%s\nCache maybe be out of date. Delete cache and retry. See %s for help.' % (e, help_url))
 
43
 
44
 
45
  def yolov5s(pretrained=False, channels=3, classes=80):
 
37
  state_dict = {k: v for k, v in state_dict.items() if model.state_dict()[k].shape == v.shape} # filter
38
  model.load_state_dict(state_dict, strict=False) # load
39
  return model
40
+
41
  except Exception as e:
42
  help_url = 'https://github.com/ultralytics/yolov5/issues/36'
43
+ s = 'Cache maybe be out of date, deleting cache and retrying may solve this. See %s for help.' % help_url
44
+ raise Exception(s) from e
45
 
46
 
47
  def yolov5s(pretrained=False, channels=3, classes=80):
requirements.txt CHANGED
@@ -1,13 +1,13 @@
1
  # pip install -U -r requirements.txt
2
  Cython
3
- numpy==1.17.3
4
  opencv-python
5
  torch>=1.5.1
6
  matplotlib
7
  pillow
8
  tensorboard
9
  PyYAML>=5.3
10
- torchvision
11
  scipy
12
  tqdm
13
  git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI
 
1
  # pip install -U -r requirements.txt
2
  Cython
3
+ numpy>=1.18.5
4
  opencv-python
5
  torch>=1.5.1
6
  matplotlib
7
  pillow
8
  tensorboard
9
  PyYAML>=5.3
10
+ torchvision>=0.6
11
  scipy
12
  tqdm
13
  git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI