Merge pull request #56 from channeladam/master

Fix initial basic setup / new starter issues
ultralytics · Jun 14, 2020 · 9bf35e7 · 9bf35e7
2 parents 93bbf1b + 335c607
commit 9bf35e7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,14 +1,16 @@
 # pip install -U -r requirements.txt
-numpy==1.17
+Cython
+numpy
 opencv-python
 torch >= 1.5
 matplotlib
-pycocotools
-tqdm
 pillow
 tensorboard
 pyyaml
-
+torchvision
+scipy
+tqdm
+'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
 
 # Nvidia Apex (optional) for mixed precision training --------------------------
 # git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex

diff --git a/weights/download_weights.sh b/weights/download_weights.sh
@@ -4,4 +4,5 @@
 python3 -c "from utils.google_utils import *;
 attempt_download('weights/yolov5s.pt');
 attempt_download('weights/yolov5m.pt');
-attempt_download('weights/yolov5l.pt')"
+attempt_download('weights/yolov5l.pt');
+attempt_download('weights/yolov5x.pt')"