diff --git a/utils/datasets.py b/utils/datasets.py index 083b26a..028e12f 100644 --- a/utils/datasets.py +++ b/utils/datasets.py @@ -179,13 +179,13 @@ class LoadImages: # for inference self.count += 1 img0 = cv2.imread(path) # BGR # mask - site = np.array([[[0, 1024], [0, 571], [313, 365], [949, 367], [1277, 596], [1280, 1024]]], dtype=np.int32) - im = np.zeros(img0.shape[:2], dtype="uint8") - cv2.polylines(im, site, 1, 255) - cv2.fillPoly(im, site, 255) - mask = im - masked = cv2.bitwise_or(img0, img0, mask=mask) - img0 = masked + # site = np.array([[[0, 1024], [0, 571], [313, 365], [949, 367], [1277, 596], [1280, 1024]]], dtype=np.int32) + # im = np.zeros(img0.shape[:2], dtype="uint8") + # cv2.polylines(im, site, 1, 255) + # cv2.fillPoly(im, site, 255) + # mask = im + # masked = cv2.bitwise_or(img0, img0, mask=mask) + # img0 = masked # cv2.namedWindow('mask', cv2.WINDOW_NORMAL) # cv2.imshow("mask", img0) # cv2.waitKey(0)