コード例 #1
0
#     project_list = (
#                 'nyu_sornet_resnet_decoder_masked_pretrained_L1_12k',
#                 'nyu_resnet50_sornet_pt_m_L2_12k',
#                 'nyu_sornet_resnet_decoder_masked_pretrained_L1_230k',
#                 'nyu_sornet_resnet_decoder_masked_pt_L1_12k',
#                 'train_nyu_resnet152_sornet_m_pt_L1_230',
#                 'nyu_resnet50_sornet_pt_m_berhu_deltaLaina_12k',
# )
# el

if opt.name == '':
    # project_list = sorted(os.listdir(opt.checkpoints_path))
    project_list = sorted(
        glob.glob(
            os.path.join(opt.checkpoints_path, '*{}*'.format(opt.s_string))))
else:
    project_list = [opt.name]

for opt.name in project_list:
    visualizer = Visualizer(opt)
    is_load_data, is_load_data_val = load_plot_data(opt)
    if opt.train and is_load_data:
        visualizer.display_existing_plot(visualizer.plot_data, 'train')
    if opt.validate and is_load_data_val:
        visualizer.display_existing_plot(visualizer.plot_data_val, 'val')
    if is_load_data or is_load_data_val:
        print('Data from {}'.format(os.path.basename(opt.name)))
    #if not is_load_data and not is_load_data_val:
    #    print('NO Data from {}'.format(opt.name))
    opt.display_id += 20