Skip to content

Commit

Permalink
Add keras_application_tests core.
Browse files Browse the repository at this point in the history
Signed-off-by: Jay Zhang <[email protected]>
  • Loading branch information
fatcat-z committed Jun 6, 2024
1 parent 5c2edd1 commit ece9a22
Show file tree
Hide file tree
Showing 6 changed files with 79 additions and 79 deletions.
76 changes: 76 additions & 0 deletions .github/workflows/keras_application_test_core.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,76 @@
name: Keras2Onnx Application Tests (Core)

on:
pull_request:
push:
branches:
- main

jobs:
run_tests:
strategy:
matrix:
tf_version: ['2.9.0', '2.13.0']
python_version: ['3.8', '3.9']
opset_version: ['18', '15']
ort_version: ['1.16.3']
onnx_version: ['1.15.0']

runs-on: ubuntu-latest

steps:
- name: Set up Python (${{ matrix.python_version }})
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python_version }}

- name: Checkout code
uses: actions/checkout@v4

- name: Install dependencies (tf v${{ matrix.tf_version }})
shell: bash
run: |
python -m pip install --upgrade pip
# conda config --set always_yes yes --set changeps1 no
pip install onnxconverter-common
pip install onnx==${{ matrix.onnx_version }}
pip uninstall -y protobuf
pip install "protobuf~=3.20"
pip install h5py==3.7.0
pip install parameterized
pip install timeout-decorator
pip install coloredlogs flatbuffers
pip install tensorflow==${{ matrix.tf_version }}
pip install onnxruntime==${{ matrix.ort_version }}
pip install Pillow==8.2.0
pip install opencv-python
pip install tqdm
pip install keras-segmentation==0.2.0
git clone https://github.com/matterport/Mask_RCNN
cd Mask_RCNN
pip install -r requirements.txt
python setup.py install
cd ..
pip install matplotlib
git clone https://github.com/qqwweee/keras-yolo3
pip install keras-resnet
pip install git+https://www.github.com/keras-team/keras-contrib.git
pip install keras-tcn==2.8.3
pip install git+https://github.com/qubvel/efficientnet
pip install transformers==4.2.0
pip install keras-self-attention
pip install pytest pytest-cov pytest-runner
pip freeze --all
- name: Run keras application tests
run: |
set -x
status=0
pip install -e .
python -c "import onnxruntime"
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
cd tests/keras2onnx_applications/nightly_build
python run_all_v2.py
exit $status
2 changes: 1 addition & 1 deletion .github/workflows/pretrained_model_tests_core.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ jobs:
shell: bash
run: |
chmod +x ./tests/setup_test_env.sh
./tests/setup_test_env.sh ${{ matrix.tf_version }} ${{ matrix.ort_version }}
./tests/utils/setup_test_env.sh ${{ matrix.tf_version }} ${{ matrix.ort_version }}
- name: Fix Paths (Windows only)
if: runner.os == 'Windows'
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/pylint.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ jobs:
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: 3.8 # Specify the desired Python version (e.g., 3.8, 3.9)
python-version: 3.9 # Specify the desired Python version (e.g., 3.8, 3.9)

- name: Install dependencies
run: pip install pylint==2.4.4
Expand Down
76 changes: 0 additions & 76 deletions .github/workflows/unit_test_matrix.yml

This file was deleted.

2 changes: 1 addition & 1 deletion .github/workflows/unit_tests_core.yml
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ jobs:
shell: bash
run: |
chmod +x ./tests/setup_test_env.sh
./tests/setup_test_env.sh ${{ matrix.tf_version }} ${{ matrix.ort_version }}
./tests/utils/setup_test_env.sh ${{ matrix.tf_version }} ${{ matrix.ort_version }}
- name: Fix Paths (Windows only)
if: runner.os == 'Windows'
Expand Down
File renamed without changes.

0 comments on commit ece9a22

Please sign in to comment.