Commit
·
be63538
1
Parent(s):
bc7cdd6
Update display.py
Browse files- display.py +2 -1
display.py
CHANGED
|
@@ -9,7 +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 |
-
|
|
|
|
| 13 |
scaled_anchors = config.SCALED_ANCHORS
|
| 14 |
transforms = A.Compose(
|
| 15 |
[
|
|
|
|
| 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 |
[
|