FCN图像语义分割
FCN-8s表示从8倍下采样的特征图通过上采样恢复到原图尺寸的网络结构。现在下载数据集,这种8倍的比较好。
from download import download
url = “https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/notebook/datasets/dataset_fcn8s.tar”
download(url, “./dataset”, kind=“tar”, replace=True)
这分割用途的数据集不同,是一些随机大小,截取,像素点分类的参数
原始图像看看,就是这样的图片,只是像素大小不一,需要转换
上面是fcn网络按照论文拼出来。使用vgg16做骨架,所以下载vgg16
from download import download
from mindspore import load_checkpoint, load_param_into_net
url = “https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/notebook/datasets/fcn8s_vgg16_pretrain.ckpt”
download(url, “fcn8s_vgg16_pretrain.ckpt”, replace=True)
def load_vgg16():
ckpt_vgg16 = “fcn8s_vgg16_pretrain.ckpt”
param_vgg = load_checkpoint(ckpt_vgg16)
load_param_into_net(net, param_vgg)
评价模型质量的代码也可以拼好
模型的保存设定,训练的模型保存到ckpt目录
最后eval的时候得到准确率0。97还可以啊。
上面一行是原始图,下面是分割的。