Update app.py
Browse files
app.py
CHANGED
@@ -17,16 +17,16 @@ def process_single_img(img_bgr, model_name):
|
|
17 |
palette_dict[idx] = each[1]
|
18 |
|
19 |
if model_name == 'Mask2Former':
|
20 |
-
config_file = 'CVRP_configs/
|
21 |
checkpoint_file = 'checkpoint/Mask2Former.pth'
|
22 |
elif model_name == 'KNet':
|
23 |
-
config_file = 'CVRP_configs/
|
24 |
checkpoint_file = 'checkpoint/KNet.pth'
|
25 |
elif model_name == 'DeepLabV3+':
|
26 |
-
config_file = 'CVRP_configs/
|
27 |
checkpoint_file = 'checkpoint/DeepLabV3plus.pth'
|
28 |
elif model_name == 'Segformer':
|
29 |
-
config_file = 'CVRP_configs/
|
30 |
checkpoint_file = 'checkpoint/Segformer.pth'
|
31 |
else:
|
32 |
return None, None
|
|
|
17 |
palette_dict[idx] = each[1]
|
18 |
|
19 |
if model_name == 'Mask2Former':
|
20 |
+
config_file = 'CVRP_configs/CVRP_mask2former.py'
|
21 |
checkpoint_file = 'checkpoint/Mask2Former.pth'
|
22 |
elif model_name == 'KNet':
|
23 |
+
config_file = 'CVRP_configs/CVRP_knet.py'
|
24 |
checkpoint_file = 'checkpoint/KNet.pth'
|
25 |
elif model_name == 'DeepLabV3+':
|
26 |
+
config_file = 'CVRP_configs/CVRP_deeplabv3plus.py'
|
27 |
checkpoint_file = 'checkpoint/DeepLabV3plus.pth'
|
28 |
elif model_name == 'Segformer':
|
29 |
+
config_file = 'CVRP_configs/CVRP_segformer.py'
|
30 |
checkpoint_file = 'checkpoint/Segformer.pth'
|
31 |
else:
|
32 |
return None, None
|