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
|