Update inferEval.py

scripts/inferEval.py

@@ -13,18 +13,18 @@ class Object(object):
 def main(args):
 
     infer_args = Object()
-    infer_args.metadata = "../caltech-fish-counting/data/metadata"
-    infer_args.frames = "../caltech-fish-counting/data/images"
-    infer_args.output = "../caltech-fish-counting/data/result"
+    infer_args.metadata = "../frames/metadata"
+    infer_args.frames = "../frames/images"
+    infer_args.output = "../frames/result"
     infer_args.weights = "models/v5m_896_300best.pt"
 
     config = {
-        'conf_threshold': args.conf_threshold,
-        'nms_iou': args.nms_iou,
-        'min_length': args.min_length,
-        'max_age': args.max_age,
-        'iou_threshold': args.iou_threshold,
-        'min_hits': args.min_hits
+        'conf_threshold': float(args.conf_threshold),
+        'nms_iou': float(args.nms_iou),
+        'min_length': float(args.min_length),
+        'max_age': int(args.max_age),
+        'iou_threshold': float(args.iou_threshold),
+        'min_hits': int(args.min_hits)
     }
 
     infer(infer_args, config=config)