Fixing dependencies (#189)

* Fixing importing seaborn

* Remove pandas in GH action

* Add thop to requirements

.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 pandas seaborn thop
       - name: Lint with flake8
         run: |
           # stop the build if there are Python syntax errors or undefined names

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

yolort/v5/utils/plots.py

@@ -12,7 +12,6 @@
 import matplotlib.pyplot as plt
 import numpy as np
 import pandas as pd
-import seaborn as sn
 import torch
 from PIL import Image, ImageDraw, ImageFont
 
@@ -361,7 +360,12 @@ def plot_study_txt(path="", x=None):
 
 
 def plot_labels(labels, names=(), save_dir=Path("")):
-    # plot dataset labels
+    """
+    Plot dataset labels
+    """
+
+    import seaborn as sn
+
     print("Plotting labels... ")
     c, b = labels[:, 0], labels[:, 1:].transpose()  # classes, boxes
     nc = int(c.max() + 1)  # number of classes