Niv Sardi commited on
Commit
9849035
1 Parent(s): c9f2bb9

python/augment: be more resilient

Browse files
Files changed (1) hide show
  1. 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 e:
163
- print(f'couldnt process {i}, {j}: {e}')
 
 
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,