Ankle/yolov5_ZWgra
1
0 Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

Merge pull request #56 from channeladam/master

Browse Source 
Fix initial basic setup / new starter issues
This commit is contained in:
Glenn Jocher authored and GitHub committed 2020-06-14 10:33:34 -07:00
2 parents 93bbf1b87a 335c607a24
commit 9bf35e7b86
2 files changed
+8 -5
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

requirements.txt
+6 -4
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
weights/download_weights.sh
Regular → Executable
+2 -1
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')"