zenon/master-thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Move weights to canonical location

Browse Source
This commit is contained in:
Tobias Eidelpes 2023-02-14 11:22:09 +01:00
parent f9de6912d8
commit fcb21ff0b9
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
evaluation/evaluation-end2end.ipynb
View File

@ -75,7 +75,7 @@
" for sample in pb(predictions_view):\n", " for sample in pb(predictions_view):\n",
" image = Image.open(sample.filepath)\n", " image = Image.open(sample.filepath)\n",
" w, h = image.size\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", "\n",
" detections = []\n", " detections = []\n",
" for _, row in pred.iterrows():\n", " for _, row in pred.iterrows():\n",

2
evaluation/evaluation.py
View File

@ -25,7 +25,7 @@ with fo.ProgressBar() as pb:
for sample in pb(predictions_view): for sample in pb(predictions_view):
image = Image.open(sample.filepath) image = Image.open(sample.filepath)
w, h = image.size w, h = image.size
pred = detect(sample.filepath, 'yolo.onnx', 'resnet.onnx') pred = detect(sample.filepath, '../weights/yolo.onnx', '../weights/resnet.onnx')
detections = [] detections = []
for _, row in pred.iterrows(): for _, row in pred.iterrows():