Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Update ci-testing.yml #445

Merged
merged 8 commits into from
Jul 19, 2020
Merged

Update ci-testing.yml #445

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
f59c54a
Update ci-testing.yml
glenn-jocher Jul 19, 2020
6e22941
Update ci-testing.yml
glenn-jocher Jul 19, 2020
517aa16
Update requirements.txt
glenn-jocher Jul 19, 2020
43b3d05
Update requirements.txt
glenn-jocher Jul 19, 2020
8c02ecd
Update google_utils.py
glenn-jocher Jul 19, 2020
ad9af8e
Update test.py
glenn-jocher Jul 19, 2020
0e5c069
Merge branch 'master' into glenn-jocher-patch-1
glenn-jocher Jul 19, 2020
2928c6d
Update ci-testing.yml
glenn-jocher Jul 19, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 2 additions & 3 deletions .github/workflows/ci-testing.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 2 additions & 2 deletions test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -220,8 +220,8 @@ def test(data,
cocoEval.accumulate()
cocoEval.summarize()
map, map50 = cocoEval.stats[:2] # update results ([email protected]:0.95, [email protected])
except:
print('pycocotools not evaluated')
except Exception as e:
print('ERROR: pycocotools unable to run: %s' % e)

# Return results
model.float() # for training
Expand Down
2 changes: 1 addition & 1 deletion utils/google_utils.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down