Coverage for scripts/testing.py: 0%

1import cv2 

2import numpy as np 

3import os 

4from ultralytics import YOLO 

5import shutil 

6from pathlib import Path 

7 

8model_path = "../models/seame_n/weights/best.pt" 

9 

10model = YOLO(model_path) 

11results = model.predict("../clutter/video_seame.avi", save=True, conf=0.2, imgsz=512) 

12 

13image_dir = "../clutter/signal_lab_test" 

14 

15for file_name in os.listdir(image_dir): 

16 file_path = os.path.join(image_dir, file_name) 

17 if os.path.isfile(file_path): 

18 model.predict(file_path, save=True, imgsz=512, conf=0.2) 

19 

20os.system('yolo val model="../models/seame_n/weights/best.pt" data="/home/seame/ObjectDetectionAvoidance/yolo_models/seame_training/data.yaml"') 

21