From dec9cc6c8bfb3f39701c7964a47390421fb58650 Mon Sep 17 00:00:00 2001 From: Rafael <77973453+Unbreak4ble@users.noreply.github.com> Date: Sat, 14 Oct 2023 13:03:29 -0400 Subject: [PATCH] Update eval.py fixed syntax error --- eval.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/eval.py b/eval.py index 7b6adc4..1b39079 100644 --- a/eval.py +++ b/eval.py @@ -1,3 +1,4 @@ +import sys import numpy as np import torch import torchvision @@ -7,7 +8,7 @@ from sklearn.metrics import confusion_matrix # Paths for image directory and model -EVAL_DIR=sys.argv[1]) +EVAL_DIR=sys.argv[1] EVAL_MODEL='models/mobilenetv2.pth' # Load the model for evaluation