commit

a0405u · Dec 29, 2020 · 7db7ee6 · 7db7ee6
1 parent ff9ea09
commit 7db7ee6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lab3/lab3.py b/lab3/lab3.py
@@ -104,6 +104,8 @@ def process(image, model, trt):
 
     index = predict(image, model, trt)
 
+    print(index)
+
     sub.set_title(classes[index])
     plt.axis('off')
     plt.imshow(image)