From fcb21ff0b9ce524d4feae5562c7adff2de703fae Mon Sep 17 00:00:00 2001 From: Tobias Eidelpes Date: Tue, 14 Feb 2023 11:22:09 +0100 Subject: [PATCH] Move weights to canonical location --- evaluation/evaluation-end2end.ipynb | 2 +- evaluation/evaluation.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/evaluation/evaluation-end2end.ipynb b/evaluation/evaluation-end2end.ipynb index 27cadeb..33449a9 100644 --- a/evaluation/evaluation-end2end.ipynb +++ b/evaluation/evaluation-end2end.ipynb @@ -75,7 +75,7 @@ " for sample in pb(predictions_view):\n", " image = Image.open(sample.filepath)\n", " w, h = image.size\n", - " pred = detect(sample.filepath, 'yolo.onnx', 'resnet.onnx')\n", + " pred = detect(sample.filepath, '../weights/yolo.onnx', '../weights/resnet.onnx')\n", "\n", " detections = []\n", " for _, row in pred.iterrows():\n", diff --git a/evaluation/evaluation.py b/evaluation/evaluation.py index dc15c6e..aca4734 100644 --- a/evaluation/evaluation.py +++ b/evaluation/evaluation.py @@ -25,7 +25,7 @@ with fo.ProgressBar() as pb: for sample in pb(predictions_view): image = Image.open(sample.filepath) w, h = image.size - pred = detect(sample.filepath, 'yolo.onnx', 'resnet.onnx') + pred = detect(sample.filepath, '../weights/yolo.onnx', '../weights/resnet.onnx') detections = [] for _, row in pred.iterrows():