Merge pull request #56 from channeladam/master

Fix initial basic setup / new starter issues

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

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')"