Skip to content

Commit

Permalink
Cleanup requirements.txt (#57)
Browse files Browse the repository at this point in the history
* Separate onnxruntime from requirements.txt

* Removing unused Mudules

* IPython is required

* Remove yaml dependencies in hubconf
  • Loading branch information
zhiqwang authored Feb 12, 2021
1 parent 22937ed commit ebf6b9d
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
5 changes: 3 additions & 2 deletions .github/workflows/ci_test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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 }}
Expand Down
2 changes: 1 addition & 1 deletion hubconf.py
Original file line number Diff line number Diff line change
@@ -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
3 changes: 0 additions & 3 deletions requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit ebf6b9d

Please sign in to comment.