Update app.py
Browse files
app.py
CHANGED
|
@@ -118,9 +118,9 @@ GLEEmodel_swin.eval()
|
|
| 118 |
cfg_eva02 = get_cfg()
|
| 119 |
add_deeplab_config(cfg_eva02)
|
| 120 |
add_glee_config(cfg_eva02)
|
| 121 |
-
|
| 122 |
checkpoints_eva = torch.load('GLEE/GLEE_{}.pth'.format(args.version))
|
| 123 |
-
cfg_eva02.merge_from_file(
|
| 124 |
GLEEmodel_eva02 = GLEE_Model(cfg_eva02, None, device, None, True).to(device)
|
| 125 |
GLEEmodel_eva02.load_state_dict(checkpoints_eva, strict=False)
|
| 126 |
GLEEmodel_eva02.eval()
|
|
|
|
| 118 |
cfg_eva02 = get_cfg()
|
| 119 |
add_deeplab_config(cfg_eva02)
|
| 120 |
add_glee_config(cfg_eva02)
|
| 121 |
+
conf_files_eva02 = 'GLEE/configs/EVA02.yaml'
|
| 122 |
checkpoints_eva = torch.load('GLEE/GLEE_{}.pth'.format(args.version))
|
| 123 |
+
cfg_eva02.merge_from_file(conf_files_eva02)
|
| 124 |
GLEEmodel_eva02 = GLEE_Model(cfg_eva02, None, device, None, True).to(device)
|
| 125 |
GLEEmodel_eva02.load_state_dict(checkpoints_eva, strict=False)
|
| 126 |
GLEEmodel_eva02.eval()
|