Acyclic
/
AutoETS2
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix typo

This commit is contained in:
TerenceLiu98 2024-05-03 20:30:39 +00:00
parent 6db4dbc306
commit c6b7bce2f4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/perception/base.py
View File

@ -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):