Merge pull request #56 from channeladam/master
Browse files- requirements.txt +6 -4
- weights/download_weights.sh +2 -1
requirements.txt
CHANGED
|
@@ -1,14 +1,16 @@
|
|
| 1 |
# pip install -U -r requirements.txt
|
| 2 |
-
|
|
|
|
| 3 |
opencv-python
|
| 4 |
torch >= 1.5
|
| 5 |
matplotlib
|
| 6 |
-
pycocotools
|
| 7 |
-
tqdm
|
| 8 |
pillow
|
| 9 |
tensorboard
|
| 10 |
pyyaml
|
| 11 |
-
|
|
|
|
|
|
|
|
|
|
| 12 |
|
| 13 |
# Nvidia Apex (optional) for mixed precision training --------------------------
|
| 14 |
# 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
|
|
|
|
| 1 |
# pip install -U -r requirements.txt
|
| 2 |
+
Cython
|
| 3 |
+
numpy
|
| 4 |
opencv-python
|
| 5 |
torch >= 1.5
|
| 6 |
matplotlib
|
|
|
|
|
|
|
| 7 |
pillow
|
| 8 |
tensorboard
|
| 9 |
pyyaml
|
| 10 |
+
torchvision
|
| 11 |
+
scipy
|
| 12 |
+
tqdm
|
| 13 |
+
'git+https://github.com/cocodataset/cocoapi.git#subdirectory=PythonAPI'
|
| 14 |
|
| 15 |
# Nvidia Apex (optional) for mixed precision training --------------------------
|
| 16 |
# 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
CHANGED
|
@@ -4,4 +4,5 @@
|
|
| 4 |
python3 -c "from utils.google_utils import *;
|
| 5 |
attempt_download('weights/yolov5s.pt');
|
| 6 |
attempt_download('weights/yolov5m.pt');
|
| 7 |
-
attempt_download('weights/yolov5l.pt')
|
|
|
|
|
|
| 4 |
python3 -c "from utils.google_utils import *;
|
| 5 |
attempt_download('weights/yolov5s.pt');
|
| 6 |
attempt_download('weights/yolov5m.pt');
|
| 7 |
+
attempt_download('weights/yolov5l.pt');
|
| 8 |
+
attempt_download('weights/yolov5x.pt')"
|