Update label_image.py

ducanhho2296 · Feb 11, 2024 · 8860e3f · 8860e3f
1 parent de53b05
commit 8860e3f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/utils/label_image.py b/utils/label_image.py
@@ -62,3 +62,4 @@ def label_image(building_df, image, i=1):
 #         padded_img = pad_image_to_square(img)
 #         transform = src.transform
 #         profile = src.profile
+#         profile.update(width=padded_img.shape[1], height=padded_img.shape[0], transform=transform)