|
|
@@ -100,16 +100,15 @@ class Settings:
|
|
|
},
|
|
|
"pokemon_front_face_reflect_ring_light_defect": {
|
|
|
"pth_path": "Model/pokemon_front_face_reflect_ring_light_defect.pth",
|
|
|
- "class_dict": {"1": "impact", "2": "wear", "3": "scratch", "4": "wear_and_impact",
|
|
|
- "5": "stain", "6": "damaged", "7": "impact"},
|
|
|
+ "class_dict": {"1": "damaged", "2": "impact", "3": "scratch", "4": "stain", "5": "wear"},
|
|
|
"img_size": {'width': 512, 'height': 512},
|
|
|
"confidence": 0.5,
|
|
|
"input_channels": 3,
|
|
|
},
|
|
|
"pokemon_front_face_no_reflect_ring_light_defect": {
|
|
|
"pth_path": "Model/pokemon_front_face_no_reflect_ring_light_defect.pth",
|
|
|
- "class_dict": {"1": "damaged", "2": "scratch", "3": "wear", "4": "stain", "5": "impact",
|
|
|
- "6": "wear_and_stain", "7": "wear_and_impact"}
|
|
|
+ "class_dict": {"1": "damaged", "2": "impact", "3": "impact", "4": "scratch",
|
|
|
+ "5": "stain", "6": "wear", "7": "wear_and_impact"}
|
|
|
,
|
|
|
"img_size": {'width': 512, 'height': 512},
|
|
|
"confidence": 0.5,
|
AnlaAnla