fix typo
This commit is contained in:
parent
6db4dbc306
commit
c6b7bce2f4
|
@ -29,7 +29,7 @@ from albumentations.pytorch import ToTensorV2
|
|||
OBJ_LABELS = {
|
||||
"truck": 0, "bicycle": 1, "car": 2, "motorcycle": 3,
|
||||
"train": 4, "bus": 5, "traffic sign": 6, "rider": 7, "person": 8,
|
||||
"traffic light NA": 9, "traffic light R": 10, "traffic light G": 11, "traffic light B": 12
|
||||
"traffic light NA": 9, "traffic light R": 10, "traffic light G": 11, "traffic light Y": 12
|
||||
}
|
||||
|
||||
def use_device(GPU):
|
||||
|
|
Loading…
Reference in New Issue