昇思25天学习打卡营第23天|munger85

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还可以啊。
上面一行是原始图，下面是分割的。