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

test

Browse Source
This commit is contained in:
huangtao
2022-06-24 13:35:08 +08:00
parent 2a5e0b193c
commit 445ec49d0c
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)