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)