提交 f8ab6cc0 编写于 作者: F FlyingQianMM

fix unet cityscapes pretrain weights info

上级 82f132f1
......@@ -140,7 +140,7 @@ def get_pretrain_weights(flag, class_name, backbone, save_dir):
elif flag == 'CITYSCAPES':
model_name = '{}_{}'.format(class_name, backbone)
if class_name == 'UNet':
logging.warning(warning_info.format(class_name, flag, 'IMAGENET'))
logging.warning(warning_info.format(class_name, flag, 'COCO'))
flag = 'COCO'
if class_name == 'HRNet' and backbone.split('_')[
-1] in ['W30', 'W32', 'W40', 'W48', 'W60', 'W64']:
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册