Spaces:
Runtime error
Runtime error
Niv Sardi
commited on
Commit
·
9849035
1
Parent(s):
c9f2bb9
python/augment: be more resilient
Browse files- python/augment.py +5 -4
python/augment.py
CHANGED
@@ -159,8 +159,10 @@ def process(args):
|
|
159 |
random.random(), random.random())
|
160 |
c = bb.to_centroid(img.shape)
|
161 |
anotations.append(c.to_anotation(label))
|
162 |
-
except AssertionError as
|
163 |
-
print(f'couldnt process {i}, {j}: {
|
|
|
|
|
164 |
|
165 |
try:
|
166 |
cv2.imwrite(f'{dest_images_path}/{basename}.png', img)
|
@@ -183,8 +185,7 @@ if __name__ == '__main__':
|
|
183 |
default=defaults.LOGOS_DATA_PATH,
|
184 |
help='dir containing logos')
|
185 |
parser.add_argument('--backgrounds', metavar='backgrounds', type=str,
|
186 |
-
|
187 |
-
default=defaults.IMAGES_PATH,
|
188 |
help='dir containing background plates')
|
189 |
parser.add_argument('--dst', dest='dest', type=str,
|
190 |
default=defaults.AUGMENTED_DATA_PATH,
|
|
|
159 |
random.random(), random.random())
|
160 |
c = bb.to_centroid(img.shape)
|
161 |
anotations.append(c.to_anotation(label))
|
162 |
+
except AssertionError as err:
|
163 |
+
print(f'couldnt process {i}, {j}: {err}')
|
164 |
+
except Exception as err:
|
165 |
+
print(f'error in mix pipeline: {err}')
|
166 |
|
167 |
try:
|
168 |
cv2.imwrite(f'{dest_images_path}/{basename}.png', img)
|
|
|
185 |
default=defaults.LOGOS_DATA_PATH,
|
186 |
help='dir containing logos')
|
187 |
parser.add_argument('--backgrounds', metavar='backgrounds', type=str,
|
188 |
+
default=defaults.SCREENSHOT_PATH,
|
|
|
189 |
help='dir containing background plates')
|
190 |
parser.add_argument('--dst', dest='dest', type=str,
|
191 |
default=defaults.AUGMENTED_DATA_PATH,
|