图像分割将label和得到的mask画入原图

最近参加天池的比赛，写了几行代码将mask和label画入原图，便于观察那些数据得到的效果不太好，详细的代码如下：

import os
import cv2
import numpy as np
import params

def show(img, mask, mode=‘B’):
“”"
:param img:图像矩阵，shape：（H，W，3）
:param mask: 标签矩阵，shape：（H，W）
:param mode: ‘R’,‘G’,‘B’
:return:
“”"
mask = mask.reshape(label.shape[0], label.shape[1], 1)
n = np.zeros(mask.shape)
if mode == ‘B’:
lar = np.concatenate((mask, n, n), axis = 2)
elif mode == ‘G’:
lar = np.concatenate((n, mask, n), axis = 2)
elif mode == ‘R’:
lar = np.concatenate((n, n, mask), axis = 2)
else:
return
lar = lar * 255
lar = lar + img
max = np.max(lar)
ret = lar/max
#下面两行移到函数外
# i += 1
# cv2.imwrite(params[‘mask_label’]+str(i)+’.png’, ret)
return ret*255
加入到训练过程中，可以得到如下的结果：

可以看出来有个问题，画出来的图会变暗

图像分割将label和得到的mask画入原图相关推荐

CNN图像分割简史：从R-CNN到Mask R-CNN（译）
一直想总结一下CNN在图像分割中的发展脉络(主要是R-CNN及其改进方法),看到一篇不错的英文博文<A Brief History of CNNs in Image Segmentation: ...
图像分割 | 训练集输入labe（mask）处理以及类别不均衡的处理-以segnet为例
Tags: ComputerVision 编译 src/caffe/layers/contrastive_loss_layer.cpp:56:30: error: no matching functi ...
二值mask图像 + RGB原图生成可视化分割结果；从二值mask获取分割轮廓点
可视化分割结果: import cv2 import numpy as np from tqdm import tqdm from PIL import Image from pathlib impo ...
matlab画入射系数和透射系数,反射系数和透射系数.ppt
反射系数和透射系数环境噪声控制工程内容组成绪论声学的基础知识噪声测量技术环境噪声影响评价噪声控制技术噪声控制技术应用 Chapter 2 声学基础知识声音的产生和传播声波的描 ...
【源码】二维码生成及美化
activity.xml <?xml version="1.0" encoding="utf-8"?> <android.support.co ...
Zxing生成自定义二维码样式
Zxing生成自定义二维码样式现在几乎每个app都带有二维码,但是google提供的zxing包,只给我们提供了核心的二维码算法,而开发过程中经常有自定义二维码的样式的需求,这时候我们就要自己写工具 ...
安卓实现个性彩色好看的二维码
前言二维码以前也写过好几篇,今天姑且是对以前知识的总结,顺便复习一下,圣人说过:温故知新. 还是和以前一样先看下效果,再来对着代码讲解. 看效果图: 之前呢,也写过用安卓实现二维码生成彩色的二维 ...
【源码】二维码透明底和去除白边功能
布局(activity.xml).权限(AndroidManifest.xml)和 MainActivity.java(基本一致) 跟先前的源码一致这里就不重复了,需要可以参考先前的. 飞机票:[源码 ...
Android之自定义生成彩色二维码
先导个zxing.jar包下面是xml布局 activity_main.xml <RelativeLayout xmlns:android="http://schemas.andro ...

图像分割将label和得到的mask画入原图

图像分割将label和得到的mask画入原图相关推荐

最新文章

热门文章