Update display.py

display.py

@@ -6,9 +6,11 @@ import random
 from albumentations.pytorch import ToTensorV2
 from yolov3 import YOLOV3_PL
 
-def inference(image: np.ndarray, iou_thresh: float = 0.5, thresh: float = 0.4, show_cam: bool = False, transparency: float = 0.5):
+def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75, show_cam: bool = False, transparency: float = 0.5):
     model = YOLOV3_PL()
     model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
+
+    scaled_anchors  =  config.SCALED_ANCHORS
     transforms =  A.Compose(
     [
         A.LongestMaxSize(max_size=config.IMAGE_SIZE),