update yolo model

app.py

@@ -56,7 +56,9 @@ id_categories = {
     6: 'Tire',
     7: 'Fishing net', #'Fishing net / cord',
     8: 'Easily namable',
-    9: 'Unclear'
+    9: 'Unclear',
+    10: 'Fragment',
+    11: 'Fragment'
 }
 
 
@@ -85,12 +87,11 @@ config_track = DotDict({
 logger.info('---Yolo model...')
 # Yolo has warning problems, so we set an env variable to remove it
 os.environ["VERBOSE"] = "False"
-URL_MODEL = "https://github.com/surfriderfoundationeurope/IA_Pau/releases/download/v0.1/yolov5.pt"
+URL_MODEL = "https://github.com/surfriderfoundationeurope/surfnet/releases/download/v01.2023/yolo_latest.pt"
 FILE_MODEL = "yolov5.pt"
 model_path = download_from_url(URL_MODEL, FILE_MODEL, "./models", logger)
 model_yolo = load_model(model_path, config_track.device, config_track.yolo_conf_thrld, config_track.yolo_iou_thrld)
 
-
 logger.info('---Centernet model...')
 URL_MODEL = "https://partage.imt.fr/index.php/s/sJi22N6gedN6T4q/download"
 FILE_MODEL = "mobilenet_v3_pretrained.pth"
@@ -131,7 +132,7 @@ def track(args):
     # centernet version
     if args.model_type == "yolo":
         logger.info("---Using Yolo")
-        detector = lambda frame: predict_yolo(model_yolo, frame, size=config_track.size, augment=False)
+        detector = lambda frame: predict_yolo(model_yolo, frame, size=config_track.size)
     elif args.model_type == "centernet":
         logger.info("---Using Centernet")
         detector = lambda frame: detect(frame, threshold=args.detection_threshold, model=model)

requirements.txt

@@ -1,4 +1,4 @@
-plastic-origins==2.1.1
+plastic-origins==2.2.3
 yolov5
 gradio
 osmnx