Noah Vriese
commited on
Commit
·
686c6f1
1
Parent(s):
5e7405a
Update visualizations
Browse files- 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-
|
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,
|