Merge pull request #28 from JanGerritsen/Bugfix_Adding_Tag_Twice

Bugfix: Adding same Tag twice to Caption
2023-01-24 07:12:56 -08:00 · 2023-01-24 07:12:56 -08:00 · f0e8ee55df
parent b9825a7077 44fcbb6aa7
commit f0e8ee55df
1 changed files with 1 additions and 5 deletions
--- a/data/image_train_item.py
+++ b/data/image_train_item.py
@ -105,10 +105,6 @@ class ImageCaption:
            if caption:
                caption += ", "
            caption += tag
-            
-            if caption:
-                caption += ", "
-            caption += tag

        return caption

@ -395,4 +391,4 @@ class ImageTrainItem:
            min_xy = min(x, y)
            image = image.crop((x_crop, y_crop, x_crop + min_xy, y_crop + min_xy))

-        return image
+        return image