Browse Source

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
5.0
Glenn Jocher GitHub 4 years ago
parent
commit
07a82f4d44
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 6 additions and 7 deletions
  1. +2
    -3
      .github/workflows/ci-testing.yml
  2. +1
    -1
      requirements.txt
  3. +2
    -2
      test.py
  4. +1
    -1
      utils/google_utils.py

+ 2
- 3
.github/workflows/ci-testing.yml View File

@@ -10,9 +10,9 @@ jobs:
strategy:
fail-fast: false
matrix:
os: [ubuntu-latest] # [ubuntu-latest, macos-latest, windows-latest]
os: [ubuntu-latest, macos-latest, windows-latest]
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-minutes: 50
@@ -41,7 +41,6 @@ jobs:
- name: Install dependencies
run: |
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 -q onnx
python --version

+ 1
- 1
requirements.txt View File

@@ -10,7 +10,7 @@ PyYAML>=5.3
torchvision>=0.6
scipy
tqdm
pycocotools>=2.0
# pycocotools>=2.0

# 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

+ 2
- 2
test.py View File

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

# Return results
model.float() # for training

+ 1
- 1
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" % (
id, name)
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
os.remove('cookie') if os.path.exists('cookie') else None


Loading…
Cancel
Save