Noah Vriese commited on
Commit
686c6f1
·
1 Parent(s): 5e7405a

Update visualizations

Browse files
Files changed (1) hide show
  1. data_objects.py +7 -8
data_objects.py CHANGED
@@ -16,7 +16,7 @@ class ObjectDetectionConfig(BaseModel):
16
 
17
  # Class specific inputs
18
  class_map: Dict = {0: 'license-plates', 1: 'License_Plate'}
19
- display_map: Dict = {0: 'license-plates', 1: 'License_Plate'}
20
  color_map: Dict = {0: (186, 223, 255), 1: (100, 255, 255)}
21
 
22
  class Detection:
@@ -182,15 +182,14 @@ class Detection:
182
 
183
  centroid: np.ndarray = self.centroid()[0]
184
  centroid_coords: Tuple[int, int] = (int(centroid[0]), int(centroid[1]))
 
 
 
 
 
 
185
 
186
  if draw_centroids:
187
-
188
- if centroid_color is None:
189
- centroid_color = self.color_centroid
190
- if centroid_radius is None:
191
- centroid_radius = self.radius_centroid
192
- if centroid_thickness is None:
193
- centroid_thickness = self.thickness_centroid
194
 
195
  cv2.circle(
196
  image_processed,
 
16
 
17
  # Class specific inputs
18
  class_map: Dict = {0: 'license-plates', 1: 'License_Plate'}
19
+ display_map: Dict = {0: 'license-plate', 1: 'license-plate'}
20
  color_map: Dict = {0: (186, 223, 255), 1: (100, 255, 255)}
21
 
22
  class Detection:
 
182
 
183
  centroid: np.ndarray = self.centroid()[0]
184
  centroid_coords: Tuple[int, int] = (int(centroid[0]), int(centroid[1]))
185
+ if centroid_color is None:
186
+ centroid_color = self.color_centroid
187
+ if centroid_radius is None:
188
+ centroid_radius = self.radius_centroid
189
+ if centroid_thickness is None:
190
+ centroid_thickness = self.thickness_centroid
191
 
192
  if draw_centroids:
 
 
 
 
 
 
 
193
 
194
  cv2.circle(
195
  image_processed,