✅ [Pass] git action with install cv2 lib

.github/workflows/main.yaml

@@ -19,6 +19,8 @@ jobs:
 
     - name: Install dependencies
       run: |
+        sudo apt-get update
+        sudo apt-get install -y libgl1-mesa-glx
         python -m pip install --upgrade pip
         pip install -r requirements.txt
 

tests/test_model/test_module.py

@@ -2,10 +2,10 @@ import sys
 from pathlib import Path
 
 import torch
-from torch import nn
 
 project_root = Path(__file__).resolve().parent.parent.parent
 sys.path.append(str(project_root))
+
 from yolo.model.module import SPPELAN, ADown, CBLinear, Conv, Pool
 
 STRIDE = 2

tests/test_model/test_yolo.py

@@ -37,7 +37,7 @@ def cfg() -> Config:
 @pytest.fixture
 def model(cfg: Config):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-    model = create_model(cfg.model)
+    model = create_model(cfg.model, weight_path=None)
     return model.to(device)
 
 

tests/test_utils/test_dataaugment.py

@@ -1,7 +1,6 @@
 import sys
 from pathlib import Path
 
-import pytest
 import torch
 from PIL import Image
 from torchvision.transforms import functional as TF

yolo/tools/data_loader.py

@@ -5,7 +5,6 @@ from threading import Event, Thread
 from typing import Generator, List, Tuple, Union
 
 import cv2
-import hydra
 import numpy as np
 import torch
 from loguru import logger
@@ -257,11 +256,6 @@ class StreamDataLoader:
             self.process_frame(frame)
         cap.release()
 
-    def cv2_to_tensor(self, frame: np.ndarray) -> Tensor:
-        frame_rgb = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
-        frame_float = frame_rgb.astype("float32") / 255.0
-        return torch.from_numpy(frame_float).permute(2, 0, 1)[None]
-
     def process_frame(self, frame):
         if isinstance(frame, np.ndarray):
             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)