mirrors
/
yolov5
mirror of https://github.com/ultralytics/yolov5.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update ci-testing.yml (#445) 

* Update ci-testing.yml

* Update ci-testing.yml

* Update requirements.txt

* Update requirements.txt

* Update google_utils.py

* Update test.py

* Update ci-testing.yml
pull/454/head
Glenn Jocher 2020-07-19 15:44:11 -07:00 committed by GitHub
parent 4102fcc9a7
commit 07a82f4d44
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
.github/workflows/ci-testing.yml vendored
View File

@ -10,9 +10,9 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
os: [ubuntu-latest] # [ubuntu-latest, macos-latest, windows-latest] os: [ubuntu-latest, macos-latest, windows-latest]
python-version: [3.8] python-version: [3.8]
model: ['yolov5s', 'yolov5m', 'yolov5l', 'yolov5x'] # models to test model: ['yolov5s'] # models to test
# Timeout: https://stackoverflow.com/a/59076067/4521646 # Timeout: https://stackoverflow.com/a/59076067/4521646
timeout-minutes: 50 timeout-minutes: 50
@ -41,7 +41,6 @@ jobs:
- name: Install dependencies - name: Install dependencies
run: | run: |
python -m pip install --upgrade pip python -m pip install --upgrade pip
pip install -q numpy # for cocoapi proper install
pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html
pip install -q onnx pip install -q onnx
python --version python --version

2
requirements.txt
View File

@ -10,7 +10,7 @@ PyYAML>=5.3
torchvision>=0.6 torchvision>=0.6
scipy scipy
tqdm tqdm
pycocotools>=2.0 # pycocotools>=2.0
# 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

4
test.py
View File

@ -220,8 +220,8 @@ def test(data,
cocoEval.accumulate() cocoEval.accumulate()
cocoEval.summarize() cocoEval.summarize()
map, map50 = cocoEval.stats[:2] # update results (mAP@0.5:0.95, mAP@0.5) map, map50 = cocoEval.stats[:2] # update results (mAP@0.5:0.95, mAP@0.5)
except: except Exception as e:
print('pycocotools not evaluated') print('ERROR: pycocotools unable to run: %s' % e)
# Return results # Return results
model.float() # for training model.float() # for training

2
utils/google_utils.py
View File

@ -51,7 +51,7 @@ def gdrive_download(id='1n_oKgR81BJtqk75b00eAjdv03qVCQn2f', name='coco128.zip'):
s = "curl -Lb ./cookie \"drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=%s\" -o %s" % ( s = "curl -Lb ./cookie \"drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=%s\" -o %s" % (
id, name) id, name)
else: # small file else: # small file
s = "curl -s -L -o %s 'drive.google.com/uc?export=download&id=%s'" % (name, id) s = 'curl -s -L -o %s "drive.google.com/uc?export=download&id=%s"' % (name, id)
r = os.system(s) # execute, capture return values r = os.system(s) # execute, capture return values
os.remove('cookie') if os.path.exists('cookie') else None os.remove('cookie') if os.path.exists('cookie') else None