mmsegmentation默认将预测得到的mask覆盖在原始图片上进行显示或保存,为了直接输出灰度图,需要对源码进行修改 修改位置mmseg/models/segmentors/base/show_result defshow_result(self,img,result,palette=None,win_name='',show=False,wait_time=0,out_file=None,opacity=0.5):img=mmcv.imread(img)img=img.copy()s...
由于我的实验都是基于MMSeg再autodl云平台做的,训练好模型后使用demo/image_demo_with_inference.py获得的推理图会带有标签,如下 解决办法 MMSeg版本为1.2.1 使用demo/image_demo.py推理,将show_result_pyplot的show参数设为false,这是因为云服务器场景不能像windows那样跳出一个框展示推理图 把show_result_pyplot的...
result = inference_model(model, img) # 在新的窗口可视化结果 show_result_pyplot(model, img, result, show=True) # 或者将可视化结果保存到图像文件夹中 # 您可以修改分割 map 的透明度 (0, 1]. show_result_pyplot(model, img, result, show=True, out_file='result.jpg', opacity=0.5) # 在一段...
plt.imshow(img_bgr[:,:,::-1]) plt.show() result = inference_model(model, img_bgr) result.keys() pred_mask = result.pred_sem_seg.data[0].cpu().numpy() pred_mask.shape np.unique(pred_mask) plt.figure(figsize=(8, 8)) plt.imshow(pred_mask) save_path='./outputs/spaceNet50_'+...
# 您可以修改分割 map的透明度(0,1].show_result_pyplot(model,img,result,show=True,out_file='result.jpg',opacity=0.5)# 在一段视频上测试并可视化分割结果 video=mmcv.VideoReader('video.mp4')forframeinvideo:result=inference_segmentor(model,frame)show_result_pyplot(model,result,wait_time=1) ...
# # windows操作系统# plt.rcParams['font.sans-serif']=['SimHei'] # 用来正常显示中文标签# plt.rcParams['axes.unicode_minus']=False # 用来正常显示负号# Mac操作系统,参考 https://www.ngui.cc/51cto/show-727683.html# 下载 simhei.ttf 字体文件# !wget https://zihao-openmmlab.obs.cn-east-3...
show_result(img, result, show=True) # or save the visualization results to image files # you can change the opacity of the painted segmentation map in (0, 1]. model.show_result(img, result, out_file='result.jpg', opacity=0.5) # test a video and show the results video = mmcv....
from mmseg.apis import inference_segmentor, init_segmentor, show_result_pyplot from mmseg.core.evaluation import get_palette config_file = 'configs/pspnet/pspnet_r50-d8_512x1024_40k_cityscapes.py' checkpoint_file = 'checkpoints/pspnet_r50-d8_512x1024_40k_cityscapes_20200605_003338-2966598c.pth'...
# single-gpu testing python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] [--show] # multi-gpu testing ./tools/dist_test.sh ${CONFIG_FILE} ${CHECKPOINT_FILE} ${GPU_NUM} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] ...
EncoderDecoder类的show_result函数位于BaseSegmentor类中,如下 classBaseSegmentor(BaseModule,metaclass=ABCMeta):def__init__(self,init_cfg=None):super(BaseSegmentor,self).__init__(init_cfg)self.fp16_enabled=Falsedefshow_result(self,img,result,palette=None,win_name='',show=False,wait_time=0,out...