diff --git a/.github/workflows/ci_test.yml b/.github/workflows/ci_test.yml index 71ee2a20..58369ab2 100644 --- a/.github/workflows/ci_test.yml +++ b/.github/workflows/ci_test.yml @@ -46,8 +46,6 @@ jobs: pip install opencv-python pip install pycocotools>=2.0.2 pip install onnxruntime - # requirements for ultralytics/yolov5 - pip install seaborn thop - name: Lint with flake8 run: | # stop the build if there are Python syntax errors or undefined names diff --git a/requirements.txt b/requirements.txt index b2010605..af9f6bd7 100644 --- a/requirements.txt +++ b/requirements.txt @@ -31,3 +31,4 @@ torchmetrics # extras -------------------------------------- # pycocotools on PyPI needs python3.7 as minimal # pycocotools>=2.0.2 # corresponds to https://github.com/ppwwyyxx/cocoapi +thop # FLOPs computation