![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
图像语义分割
通往盛夏
这个作者很懒,什么都没留下…
展开
-
【PyTorch图像语义分割】3. 训练规则:损失函数、优化
损失函数:交叉熵:对每个像素使用交叉熵。优化:略。Point:label是和输入相同大小的图像,不同的类用不同的颜色(或灰度,如1-21类分别用0-20灰度值表示)表示。FCN最终(应该)输出与输入相同大小的prediction map,每个像素应与label相对应。但是FCN的conv层的存在势必回减小最终输入的prediction map的大小,故需要deconv层或直接双线...原创 2018-12-05 14:31:42 · 3448 阅读 · 1 评论 -
【PyTorch图像语义分割】4. 使用训练好的模型测试
使用训练好的模型测试新图片1. 图像的加载2. 用网络forward测试1. 图像的加载测试图像的加载仍然是通过继承torch.ultis.data.Dataset加载。在加载训练图像的时候用的是class UAVDataSet(torch.utils.data.Dataset):需要可以返回数据、标签。但是在测试新图像的时候没有标签,故只需要返回数据就行,代码如下:class UAV...原创 2018-12-12 20:12:32 · 5826 阅读 · 5 评论