Skip to content

Commit cc5ab7b

Browse files
committed
updated the labels comment
1 parent 8dec51a commit cc5ab7b

File tree

1 file changed

+3
-2
lines changed
  • AppliedAI/openvino/udacity-intel-edge/lesson3

1 file changed

+3
-2
lines changed

AppliedAI/openvino/udacity-intel-edge/lesson3/app.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,8 +50,9 @@ def get_args():
5050

5151
# initialize the list of class labels MobileNet SSD was trained to
5252
# detect, then generate a set of bounding box colors for each class
53-
# classes for mobilenet, i had to swap car and bird
54-
# have to sawp aeroplane and person
53+
# classes for mobilenet
54+
# TODO: not sure if these classes are correct.
55+
# Need to externalize this to a config/mapping file
5556
CLASSES = ["background", "person", "bicycle", "car", "boat",
5657
"aeroplane", "bus", "bird", "cat", "chair", "cow", "diningtable",
5758
"dog", "horse", "motorbike", "bottle", "pottedplant", "sheep",

0 commit comments

Comments
 (0)