improved fruits and background merging

mcreng · Jan 14, 2019 · c0e8896 · c0e8896
1 parent 4f4cb5f
commit c0e8896
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/im_generator.py b/im_generator.py
@@ -136,7 +136,7 @@ def get_bounding_box(img):
     ms_colors = [(0, 187, 255), (0, 187, 124),
                  (241, 161, 0), (20, 83, 246)]  # in BGR
     color_idx = np.random.randint(0, 4)  # [0, 4)
-    output[output[:, :, 3] == 0] = (*ms_colors[color_idx], 255)
+    output[output[:, :, 3] <= 100] = (*ms_colors[color_idx], 255)
     return output, annos