RashiAgarwal
commited on
Commit
·
f0b653f
1
Parent(s):
be63538
Update display.py
Browse files- 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 |
[
|