27 lines
479 B
Python
27 lines
479 B
Python
ind2labels_dict = {
|
|
0: "Background",
|
|
1: "Hat",
|
|
2: "Hair",
|
|
3: "Sunglasses",
|
|
4: "Upper-clothes",
|
|
5: "Skirt",
|
|
6: "Pants",
|
|
7: "Dress",
|
|
8: "Belt",
|
|
9: "Left-shoe",
|
|
10: "Right-shoe",
|
|
11: "Face",
|
|
12: "Left-leg",
|
|
13: "Right-leg",
|
|
14: "Left-arm",
|
|
15: "Right-arm",
|
|
16: "Bag",
|
|
17: "Scarf"
|
|
}
|
|
|
|
# Calculate the reverse dict
|
|
labels2ind_dict = {v: k for k, v in ind2labels_dict.items()}
|
|
|
|
|
|
relevant_inds = [1,2,3,4,5,6,7,8,9,10,16,17]
|