From 2a5e0b193c480b1ee49e98a9e60a993e521cef91 Mon Sep 17 00:00:00 2001 From: huangtao Date: Thu, 23 Jun 2022 15:04:30 +0800 Subject: [PATCH] 0.55 --- utils/datasets.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) 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)