Guill-Bla commited on
Commit
efb89bf
·
verified ·
1 Parent(s): cd9b6a5

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. tasks/image.py +4 -0
tasks/image.py CHANGED
@@ -6,6 +6,10 @@ from sklearn.metrics import accuracy_score
6
  import random
7
  import os
8
 
 
 
 
 
9
  import os
10
  import torch
11
  import numpy as np
 
6
  import random
7
  import os
8
 
9
+ from ultralytics import YOLO
10
+ from .utils.evaluation import ImageEvaluationRequest
11
+ from .utils.emissions import tracker, clean_emissions_data, get_space_info
12
+
13
  import os
14
  import torch
15
  import numpy as np