发新帖

在Pascal Voc数据集上训练图像分割网络

[复制链接]
46 2

快来加入 TensorFlowers 大家庭!

您需要 登录 才可以下载或查看,没有帐号?加入社区

x
我尝试在Pascol Voc 2012数据集上面训练图像分割模型,可是尝试了半个月依然没有成功。

由于这个数据集内的类别不平衡什么的原因,训练的结果一直很差。 想请教一下各位有没有类似的经历,有什么心得可以分享一下的么? 求救 ~

代码: https://github.com/shenshutao/image_segmentation, 在马的数据集(数据集比较简单)上面可以训练出来结果 main.py,但是在pascal voc数据集上面一直训练不好 main_pascal.py .

我知道答案 回答被采纳将会获得10 金币 + 100 金币 已有2人回答
本楼点评(0) 收起

精彩评论2

树涛  TF荚荚  发表于 7 天前 | 显示全部楼层
正在下载ImageNet图片集,打算把encoder网络pretrain一下试试。
本楼点评(0) 收起
树涛  TF荚荚  发表于 7 天前 | 显示全部楼层
是我模型写错了... 最后那层softmax忘记加了,坑了自己 ...
本楼点评(0) 收起
您需要登录后才可以回帖 登录 | 加入社区

本版积分规则

主题

帖子

16

积分
快速回复 返回顶部 返回列表