diff --git a/requirements.txt b/requirements.txt index 1328818..d0d0c09 100755 --- 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 old mode 100644 new mode 100755 index d799230..6834ddb --- 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')"