Update engine.py

engine.py

@@ -46,6 +46,8 @@ class Engine(f.Pipe):
         assert len(predictor.results) == 1, 'Only support batch size 1'
         preds = predictor.results[0]
         self.stamp_detector.model.to("cuda:1")
+        print(preds.orig_img)
+        print(preds.boxes.xyxy)
         remove_idc = remove_potiential_table_fp(self.stamp_detector, preds.orig_img, preds.boxes.xyxy, self.cfg.STAMP_REMOVING_IOU_THRESHOLD)
         stamp_removed_preds = remove_box_by_idc(preds.boxes, remove_idc)
         predictor.results[0].boxes = stamp_removed_preds