diff --git a/.github/workflows/ci-testing.yml b/.github/workflows/ci-testing.yml index eb0e9483d152..3422d205fbe4 100644 --- a/.github/workflows/ci-testing.yml +++ b/.github/workflows/ci-testing.yml @@ -50,8 +50,9 @@ jobs: - name: Install dependencies run: | python -m pip install --upgrade pip - pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html \ - onnx tensorflow-cpu # wandb + # tensorflowjs for unit test + pip install -qr requirements.txt -f https://download.pytorch.org/whl/cpu/torch_stable.html tensorflowjs \ + onnx tensorflow-cpu # wandb python --version pip --version pip list @@ -83,7 +84,7 @@ jobs: # Export python models/yolo.py --cfg ${{ matrix.model }}.yaml # build PyTorch model python models/tf.py --weights ${{ matrix.model }}.pt # build TensorFlow model - python export.py --weights ${{ matrix.model }}.pt --img 64 --include torchscript onnx # export + pytest tests --weights $weights # Python python - < t.List: formats = export_formats() return formats[formats['Format'].isin(( 'ONNX', - 'OpenVINO', + # Numpy version mismatch between + # OpenVINO and tfjs where OpenVINO requires numpy 1.20 + # and tfjs requires > 1.20 + # 'OpenVINO', 'CoreML', 'TensorFlow SavedModel', 'TensorFlow GraphDef', @@ -107,14 +110,6 @@ def test_export_cpu(weights, export_format_row: t.List): # make img small for quick tests img_sz = (160, 160) - # As of now, openvino requires numpy < 1.20. - # numpy will be downgraded during openvino run - # so we need to re-upgrade numpy - if export_format_argument == 'openvino': - subprocess.run('pip install numpy==1.19.5', shell=True) - elif export_format_argument == 'tfjs': - subprocess.run('pip install --upgrade numpy', shell=True) - # create the model run(weights=weights, imgsz=img_sz, include=(export_format_argument,), int8=True) output_path = weights.replace('.pt', suffix.lower())