Align the label index in LABEL_MAP to match cls_id (#1244)

deepchecks · Apr 11, 2022 · 0475894 · 0475894
1 parent 3b74680
commit 0475894
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/tutorials/vision/plot_detection_tutorial.py b/docs/source/tutorials/vision/plot_detection_tutorial.py
@@ -315,7 +315,7 @@ def infer_on_batch(self, batch, model, device):
 # We have a single label here, which is the tomato class
 # The label_map is a dictionary that maps the class id to the class name, for display purposes.
 LABEL_MAP = {
-    0: 'Tomato'
+    1: 'Tomato'
 }
 training_data = TomatoData(data_loader=train_loader, label_map=LABEL_MAP)
 val_data = TomatoData(data_loader=val_loader, label_map=LABEL_MAP)