diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml index 813c9ad0..0e92fb9d 100644 --- a/.github/workflows/ci_test.yml +++ b/.github/workflows/ci_test.yml @@ -39,8 +39,9 @@ jobs: pip install flake8 pytest pip install pytest-cov if [ -f requirements.txt ]; then pip install -r requirements.txt; fi - pip install opencv-python - pip install pycocotools>=2.0.2 + pip install -U opencv-python + pip install -U pycocotools>=2.0.2 + pip install -U onnxruntime - name: Install PyTorch ${{ matrix.torch }} Version run: | pip install ${{ matrix.pip_address }} diff --git a/hubconf.py b/hubconf.py index 351ee600..d90d4cf6 100644 --- a/hubconf.py +++ b/hubconf.py @@ -1,4 +1,4 @@ # Optional list of dependencies required by the package -dependencies = ['yaml', 'torch', 'torchvision'] +dependencies = ['torch', 'torchvision'] from yolort.models import yolov5s, yolov5m, yolov5l diff --git a/requirements.txt b/requirements.txt index 7d14536f..ae7c1b81 100644 --- a/requirements.txt +++ b/requirements.txt @@ -5,12 +5,9 @@ Cython matplotlib>=3.2.2 numpy>=1.18.5 pillow -PyYAML>=5.3 scipy>=1.4.1 -tensorboard>=2.2 tqdm>=4.41.0 onnx>=1.8.0 -onnxruntime>=1.6.0 ipython # pycocotools need python3.7 as minimal # pip install -U pycocotools>=2.0.2 # corresponds to https://github.com/ppwwyyxx/cocoapi