Update gen_landmark_aug_12.py

JellyDream · Sep 4, 2017 · 7107251 · 7107251
1 parent 4437e8e
commit 7107251
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/prepare_data/gen_landmark_aug_12.py b/prepare_data/gen_landmark_aug_12.py
@@ -108,7 +108,7 @@ def GenerateData(ftxt, output,net,argument=False):
                 resized_im = cv2.resize(cropped_im, (size, size))
                 #cal iou
                 iou = IoU(crop_box, np.expand_dims(gt_box,0))
-                if iou > -1:
+                if iou > 0.65:
                     F_imgs.append(resized_im)
                     #normalize
                     for index, one in enumerate(landmarkGt):