CVRPDataset commited on
Commit
d717fc6
·
verified ·
1 Parent(s): b5b44fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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/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
 
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