Update dataquality/utils/semantic_segmentation/errors.py

rungalileo · May 22, 2023 · b0be115 · b0be115
1 parent 004e97d
commit b0be115
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dataquality/utils/semantic_segmentation/errors.py b/dataquality/utils/semantic_segmentation/errors.py
@@ -208,7 +208,7 @@ def calculate_dep_polygons_batch(
 
     for idx in range(len(resized_dep_maps)):
         dep_map = resized_dep_maps[idx]
-        gold_polygons = polygons_batch[idx]
-        for polygon in gold_polygons:
+        polygons = polygons_batch[idx]
+        for polygon in polygons:
             polygon_img = draw_polygon(polygon, dep_map.shape)
             polygon.data_error_potential = calculate_dep_polygon(dep_map, polygon_img)