Add thop to requirements

zhiqwang · Oct 4, 2021 · 3bce152 · 3bce152
1 parent bf25cd1
commit 3bce152
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git 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
@@ -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