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

Merge pull request #56 from channeladam/master 

Fix initial basic setup / new starter issues
This commit is contained in:
Glenn Jocher 2020-06-14 10:33:34 -07:00 committed by GitHub
parent 93bbf1b87a 335c607a24
commit 9bf35e7b86
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
requirements.txt
View File

@ -1,14 +1,16 @@
# pip install -U -r requirements.txt # pip install -U -r requirements.txt
numpy==1.17 Cython
numpy
opencv-python opencv-python
torch >= 1.5 torch >= 1.5
matplotlib matplotlib
pycocotools
tqdm
pillow pillow
tensorboard tensorboard
pyyaml pyyaml
torchvision
scipy
tqdm
'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
# Nvidia Apex (optional) for mixed precision training -------------------------- # 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 # 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

3
weights/download_weights.sh Normal file → Executable file
View File

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