affa2df
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
IDX2TAGs = {
0: "bicycle",
1: "bus",
2: "car",
3: "motorbike",
4: "person"
}
IDX2COLORs = {
0: "#FF5733",
1: "#6E0DD0",
2: "#B2B200",
3: "#009DFF",
4: "#FF33A8"
}
IMAGE_SIZE = (448, 448)
PATH_MODEL = "weight/best_m.onnx" |
