flake8 compliant

2024-02-24 23:42:55 +01:00
parent 993c9a122b
commit fa2230d5cf
23 changed files with 157 additions and 106 deletions
@@ -3,11 +3,12 @@ from src.database import ModelOutputs

 if __name__ == "__main__":
    # instantiate data object
-    annotation = {
-        
-    }
-    vis_com_list = [ModelOutputs.from_annotation(path) for path in img_path_list]
+    vis_com_list = [
+        ModelOutputs.from_random()
+        for i
+        in range(3)
+    ]
    # generate random predictions
    [vis_com.generate_random_prediction() for vis_com in vis_com_list]
    for vis_com in vis_com_list:
-        print(vis_com)
+        print(vis_com)