lichen/ieemoo-ai-zhanting
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

0.55

Browse Source
This commit is contained in:
huangtao
2022-06-23 15:04:30 +08:00
parent 2e12acb016
commit 2a5e0b193c
1 changed files with 7 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
utils/datasets.py
View File

@ -179,13 +179,13 @@ class LoadImages: # for inference
self.count += 1 self.count += 1
img0 = cv2.imread(path) # BGR img0 = cv2.imread(path) # BGR
# mask # mask
site = np.array([[[0, 1024], [0, 571], [313, 365], [949, 367], [1277, 596], [1280, 1024]]], dtype=np.int32) # 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") # im = np.zeros(img0.shape[:2], dtype="uint8")
cv2.polylines(im, site, 1, 255) # cv2.polylines(im, site, 1, 255)
cv2.fillPoly(im, site, 255) # cv2.fillPoly(im, site, 255)
mask = im # mask = im
masked = cv2.bitwise_or(img0, img0, mask=mask) # masked = cv2.bitwise_or(img0, img0, mask=mask)
img0 = masked # img0 = masked
# cv2.namedWindow('mask', cv2.WINDOW_NORMAL) # cv2.namedWindow('mask', cv2.WINDOW_NORMAL)
# cv2.imshow("mask", img0) # cv2.imshow("mask", img0)
# cv2.waitKey(0) # cv2.waitKey(0)