scorevision: push artifact

miner.py

@@ -48,6 +48,10 @@ class Miner:
             print(f'CUDA session creation skipped/failed, using CPU: {e}')
             self.session = ort.InferenceSession(str(model_path), sess_options=sess_options, providers=['CPUExecutionProvider'])
         print('ORT session providers:', self.session.get_providers())
+        self.sink_name = 'tuktuk'
+        self.sink_margin = 0.05
+        self.sink_index = None
+        self.vehicle_cols = None
         model_class_order = self._read_model_class_order()
         if model_class_order is None:
             model_class_order = list(self._model_class_order)
@@ -92,10 +96,6 @@ class Miner:
         self.square_pad_frac = 0
         self.dual_label_pair = ('bus', 'truck')
         self.dual_label_below = 0
-        self.sink_name = 'tuktuk'
-        self.sink_margin = 0.05
-        self.sink_index = None
-        self.vehicle_cols = None
         self._avg_iou = float(np.mean(self._iou_thres_array))
         print(f'ONNX model loaded from: {model_path}')
         print(f'ONNX input: name={self.input_name}, shape={self.input_shape}')