mmseg-1.0.0rc3新版本图片批量测试

方案一：SegLocalVisualizer_hook中间隔的默认值为 50。您可以在 mmseg/.mim/tools/test.py 中修改函数trigger_visualization_hook。添加 visualization_hook['interval'] = 1即可。

方案二：New a test.py

from mmseg.apis import inference_model, init_model, show_result_pyplot
import os
import time

def generate_pseudo_masks(config_file, checkpoint_file, dir_save_pseudo_masks,list_images):
    model = init_model(config_file, checkpoint_file, device='cuda:0')
    model.PALETTE = [[0,0,0],[128, 64, 128], [244, 35, 232], [70, 70, 70], [102, 102, 156],
                 [190, 153, 153], [153, 153, 153], [250, 170,30], [220, 220, 0],
                 [107, 142, 35], [152, 251, 152], [70, 130, 180],
                 [220, 20, 60], [255, 0, 0], [0, 0, 142], [0, 0, 70],
                 [0, 60, 100], [0, 80, 100], [0, 0, 230], [119, 11, 32],[125,0,0]]
    for filename in os.listdir(list_images):
        img = os.path.join(list_images, filename)