RashiAgarwal commited on
Commit
f0b653f
·
1 Parent(s): be63538

Update display.py

Browse files
Files changed (1) hide show
  1. display.py +2 -2
display.py CHANGED
@@ -9,8 +9,8 @@ from yolov3 import YOLOV3_PL
9
  def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75, transparency: float = 0.5):
10
  model = YOLOV3_PL()
11
  model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
12
- iou_thresh = 0.75
13
- thresh = 0.75
14
  scaled_anchors = config.SCALED_ANCHORS
15
  transforms = A.Compose(
16
  [
 
9
  def inference(image: np.ndarray, iou_thresh: float = 0.75, thresh: float = 0.75, transparency: float = 0.5):
10
  model = YOLOV3_PL()
11
  model.load_state_dict(torch.load("model.pth", map_location=torch.device('cpu')), strict=False)
12
+ # iou_thresh = 0.75
13
+ # thresh = 0.75
14
  scaled_anchors = config.SCALED_ANCHORS
15
  transforms = A.Compose(
16
  [