Skip to content

Commit d3ef2c9

Browse files
authored
Merge pull request #1 from onnx/main
update
2 parents bb0d4cf + e77240a commit d3ef2c9

File tree

232 files changed

+10558
-514
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

232 files changed

+10558
-514
lines changed

.github/ISSUE_TEMPLATE/model_request.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -24,4 +24,4 @@ Why is this model an improvement upon current ONNX model zoo offerings?
2424
### Notes
2525
Any additional information, code snippets
2626

27-
The ONNX team strongly encourages community members to submit a PR for models they want to use in the ONNX model zoo, and always welcomes new contributors. To get started, look at the [ONNX Model Zoo contribution page](https://github.com/onnx/models/blob/master/contribute.md).
27+
The ONNX team strongly encourages community members to submit a PR for models they want to use in the ONNX model zoo, and always welcomes new contributors. To get started, look at the [ONNX Model Zoo contribution page](https://github.com/onnx/models/blob/main/contribute.md).

.github/PULL_REQUEST_TEMPLATE.md

+12
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,15 @@ model_proto = onnx.load("path to .onnx file")
1212
checker.check_model(model_proto)
1313
```
1414

15+
### Contribute a Gradio Demo to ONNX Organization on Hugging Face
16+
17+
* Create an account on Hugging Face: https://huggingface.co/join
18+
* See list of models left to add to ONNX organization, please refer to the table with the [Models list](https://github.com/onnx/models#models)
19+
* Add Gradio Demo under your username, see blog for setting up Gradio Demo on Hugging Face: https://huggingface.co/blog/gradio-spaces
20+
* Request to join ONNX Organization: https://huggingface.co/onnx
21+
* Once approved transfer model from your username to ONNX organization
22+
* Add a badge for model in model table, see examples in [Models list](https://github.com/onnx/models#models)
23+
1524
## Model
1625

1726
|Model |Download | Download (with sample test data)|ONNX version|Opset version|Accuracy |
@@ -119,6 +128,9 @@ https://github.com/microsoft/onnxruntime/blob/master/tools/python/PythonTools.md
119128
120129
<hr>
121130
131+
### Update ONNX_HUB_MANIFEST.json for ONNX Hub
132+
If this PR does update/add .onnx or .tar.gz files, please use `python workflow_scripts/generate_onnx_hub_manifest.py --target diff` to update ONNX_HUB_MANIFEST.json with according model information (especially SHA) for ONNX Hub.
133+
122134
### References
123135
Link to paper or references.
124136

.github/workflows/linux_ci.yml

+17-12
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33

44
name: Linux CI
55

6-
# Triggers the workflow on push or pull request events but only for the master branch
6+
# Triggers the workflow on push or pull request events but only for the main branch
77
on:
88
push:
9-
branches: [ master ]
9+
branches: [ main ]
1010
pull_request:
11-
branches: [ master ]
11+
branches: [ main ]
1212

1313
jobs:
1414
# This workflow contains a single job called "build"
@@ -17,9 +17,8 @@ jobs:
1717
runs-on: ubuntu-latest
1818
strategy:
1919
matrix:
20-
python-version: [3.7]
20+
python-version: [3.8]
2121
architecture: ['x64']
22-
onnx-version: ['1.7']
2322

2423
steps:
2524
- uses: actions/checkout@v2
@@ -34,13 +33,19 @@ jobs:
3433
- name: Install dependencies
3534
run: |
3635
python -m pip install --upgrade pip
37-
python -m pip install onnx==${{ matrix.onnx-version }} requests
38-
python -m pip install onnxruntime
39-
- name: Test new models by onnxs
36+
python -m pip install onnx onnxruntime requests py-cpuinfo
37+
python -m cpuinfo
38+
39+
- name: Test updated ONNX_HUB_MANIFEST.json
4040
run: |
41-
python workflow_scripts/test_models.py --target onnx
42-
- name: Test new models by onnxruntime
41+
python -m pip install typepy BeautifulSoup4 markdown pandas
42+
python workflow_scripts/generate_onnx_hub_manifest.py --target diff --drop
43+
git diff --exit-code -- ONNX_HUB_MANIFEST.json || { echo 'Please use "python workflow_scripts/generate_onnx_hub_manifest.py --target diff" to update ONNX_HUB_MANIFEST.json.' ; exit 1; }
44+
45+
- name: Test new models by onnx
4346
run: |
44-
python workflow_scripts/test_models.py --target onnxruntime
47+
python workflow_scripts/test_models.py --target onnx --drop
4548
46-
49+
- name: Test new models by onnxruntime
50+
run: |
51+
python workflow_scripts/test_models.py --target onnxruntime --drop

.github/workflows/windows_ci.yml

+13-9
Original file line numberDiff line numberDiff line change
@@ -3,12 +3,12 @@
33

44
name: Windows CI
55

6-
# Triggers the workflow on push or pull request events but only for the master branch
6+
# Triggers the workflow on push or pull request events but only for the main branch
77
on:
88
push:
9-
branches: [ master ]
9+
branches: [ main ]
1010
pull_request:
11-
branches: [ master ]
11+
branches: [ main ]
1212

1313
jobs:
1414
# This workflow contains a single job called "build"
@@ -17,9 +17,9 @@ jobs:
1717
runs-on: windows-latest
1818
strategy:
1919
matrix:
20-
python-version: [3.7]
20+
python-version: [3.8]
2121
architecture: ['x64']
22-
onnx-version: ['1.7']
22+
2323
steps:
2424
- uses: actions/checkout@v2
2525
name: Checkout repo
@@ -33,11 +33,15 @@ jobs:
3333
- name: Install dependencies
3434
run: |
3535
python -m pip install --upgrade pip
36-
python -m pip install onnx==${{ matrix.onnx-version }} requests
37-
python -m pip install onnxruntime
36+
# TODO: now ONNX only supports Protobuf <= 3.20.1
37+
python -m pip install protobuf==3.20.1
38+
python -m pip install onnx onnxruntime requests py-cpuinfo
39+
python -m cpuinfo
40+
3841
- name: Test new models by onnx
3942
run: |
40-
python workflow_scripts/test_models.py --target onnx
43+
python workflow_scripts/test_models.py --target onnx --drop
44+
4145
- name: Test new models by onnxruntime
4246
run: |
43-
python workflow_scripts/test_models.py --target onnxruntime
47+
python workflow_scripts/test_models.py --target onnxruntime --drop

0 commit comments

Comments
 (0)