First model version

det_demo.py

@@ -573,7 +573,7 @@ class DetDemo(object):
             polygon = list(map(int, polygon))
         return polygon
 
-    def visualization(self, image, polygons, masks):
+    def visualization(self, image, polygons, masks, boxes):
         green = np.ones(image.shape).astype(np.uint8)
         green[...,0] = 0
         green[...,1] = 255
@@ -581,6 +581,7 @@ class DetDemo(object):
         for mask in masks:
             image[mask] = image[mask] * 0.5 + green[mask] * 0.5
         for box in boxes:
+            print(box)
             cv2.rectangle(image,box,True,(0,0,255))
         '''
         for polygon in polygons: