From 445ec49d0c78676687be901d0d1afc6ca3b38c40 Mon Sep 17 00:00:00 2001
From: huangtao <huangtao@ieemoo.com>
Date: Fri, 24 Jun 2022 13:35:08 +0800
Subject: [PATCH] test

---
 utils/datasets.py | 14 +++++++-------
 1 file changed, 7 insertions(+), 7 deletions(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 028e12f..083b26a 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)