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
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)