| """Target names shared by training and torch-free inference.""" |
|
|
| ALIASES = { |
| "bicycles": "bicycle", |
| "motorcycles": "motorcycle", |
| "cars": "car", |
| "bus": "bus", |
| "buses": "bus", |
| "crosswalks": "crosswalk", |
| "a fire hydrant": "fire hydrant", |
| "fire hydrants": "fire hydrant", |
| "stairs": "stairs", |
| "traffic lights": "traffic light", |
| "bridges": "bridge", |
| "tractors": "tractor", |
| "taxis": "taxi", |
| "hydrant": "fire hydrant", |
| "stair": "stairs", |
| "palm": "palm tree", |
| "palm trees": "palm tree", |
| "chimneys": "chimney", |
| "parking meters": "parking meter", |
| "mountains": "mountain", |
| "mountains or hills": "mountain", |
| "tlight": "traffic light", |
| } |
| |
| BACKGROUND = "other" |
|
|
|
|
| def target_name(value): |
| value = value.strip().lower() |
| return ALIASES.get(value, value) |
|
|
