您好，如果想预测出测试集图片后把所有的预测结果图片保存到一个文件夹，不是随机展示，该怎么改呢

[Wangcancan1127]

您好，想问一下如果想预测出测试集图片后把所有的预测结果图片保存到一个文件夹，不是随机展示，该怎么改呢，希望您能回复，非常感激您

[yizt]

@Wangcancan1127 在display_instances函数中，将_, ax = plt.subplots(1, figsize=figsize)改为，fig, ax = plt.subplots(1, figsize=figsize),并返回fig; 然后每个fig保存就好

[Wangcancan1127]

你好我改了
if not ax: fig, ax = plt.subplots(1, figsize=figsize) auto_show = True return fig 但是inference.py里面也要改吧 image_ids = np.random.choice(len(all_img_info), 9, replace=False) fig = plt.figure(figsize=(20, 20)) for idx, image_id in enumerate(image_ids): ax = fig.add_subplot(3, 3, idx + 1) _show_inference(image_id, ax) fig.savefig('demo_images/inferece_examples.{}.png'.format(np.random.randint(10))) 这块也需要改吧