我自己测试iou达不到你论文中那么高。

[haotianmingyue]

我就是尝试了一下RHSEDB数据集，用你的代码跑了好多次，最高才82%。是有什么特殊技巧吗？

[Asker0Lee]

并没有特殊技巧，整个算法由三个模型构成，如果每个模型都存在一些误差，最终会积累下来，不过82%确实有点低了，你可以尝试一个个训练。其次这篇文章更多是展现一种思路，因此模型架构上没有去尽力微调，每个模型都存在一定的性能余量，你可以多加一些epoch，或者施加一些简单的数据增强，整体效果会有明显提升。

[rorozi]

读明白了论文，想要复现一下代码，一时没有头绪，请问具体该怎么运行呢？（文件的顺序）vgg和content net的参数下载的两个文件该怎么使用呢？

[Asker0Lee]

下载下来放到指定目录即可，项目中有文件路径说明

[ZXX-Andrew]

我在跑的时候也有这个问题，我的IOU直接是0了，也不知道为什么跑了很多次还是这样。。。。。环境也是按照文档配置了，跑的时候一两个warning但是没有影响训练，但是最后Seg_Net和Extract_Net的结果mean_iou都是0也不知道为什么。。。用模型推理的代码跑过一遍之后也没有识别出笔画，推理的程序运行完以后就是原来的字显示了一遍没有颜色。看程序里似乎识别出了有多少个笔画，但是所有笔画位置相关的张量全部都是0。

[haotianmingyue]

我在跑的时候也有这个问题，我的IOU直接是0了，也不知道为什么跑了很多次还是这样。。。。。环境也是按照文档配置了，跑的时候一两个warning但是没有影响训练，但是最后Seg_Net和Extract_Net的结果mean_iou都是0也不知道为什么。。。用模型推理的代码跑过一遍之后也没有识别出笔画，推理的程序运行完以后就是原来的字显示了一遍没有颜色。看程序里似乎识别出了有多少个笔画，但是所有笔画位置相关的张量全部都是0。

我记得这代码是不能直接用的，你要改点东西。

[MengLi-l1]

我在跑的时候也有这个问题，我的IOU直接是0了，也不知道为什么跑了很多次还是这样。。。。。环境也是按照文档配置了，跑的时候一两个warning但是没有影响训练，但是最后Seg_Net和Extract_Net的结果mean_iou都是0也不知道为什么。。。用模型推理的代码跑过一遍之后也没有识别出笔画，推理的程序运行完以后就是原来的字显示了一遍没有颜色。看程序里似乎识别出了有多少个笔画，但是所有笔画位置相关的张量全部都是0。

SDNet有结果吗？建议检查下第三方库，有些第三方库如opencv-python执行错误时并不会报错。

[ZXX-Andrew]

我在跑的时候也有这个问题，我的IOU直接是0了，也不知道为什么跑了很多次还是这样。。。。。环境也是按照文档配置了，跑的时候一两个warning但是没有影响训练，但是最后Seg_Net和Extract_Net的结果mean_iou都是0也不知道为什么。。。用模型推理的代码跑过一遍之后也没有识别出笔画，推理的程序运行完以后就是原来的字显示了一遍没有颜色。看程序里似乎识别出了有多少个笔画，但是所有笔画位置相关的张量全部都是0。

SDNet有结果吗？建议检查下第三方库，有些第三方库如opencv-python执行错误时并不会报错。

SDNet有结果看起来还挺正常的。第三方库的话opencv就是正常用pip安装的。

[MengLi-l1]

我在跑的时候也有这个问题，我的IOU直接是0了，也不知道为什么跑了很多次还是这样。。。。。环境也是按照文档配置了，跑的时候一两个warning但是没有影响训练，但是最后Seg_Net和Extract_Net的结果mean_iou都是0也不知道为什么。。。用模型推理的代码跑过一遍之后也没有识别出笔画，推理的程序运行完以后就是原来的字显示了一遍没有颜色。看程序里似乎识别出了有多少个笔画，但是所有笔画位置相关的张量全部都是0。

SDNet有结果吗？建议检查下第三方库，有些第三方库如opencv-python执行错误时并不会报错。

SDNet有结果看起来还挺正常的。第三方库的话opencv就是正常用pip安装的。

这个代码是22年的，opencv-python库也有过几次比较大的更新，第三方库可能不兼容。主要是numpy和opencv-python，这个两个版本我记不清了，numpy应该在1.24左右，opencv-python应该是3.几，你需要自己排查下。

[ZXX-Andrew]

现在的情况就是

mean_iou约等于0，也不知道怎么回事但是出来的时候就是一片黑的，版本号现在也调好了

开始的时候是这样的就很奇怪了已经，后来iou越来越小了就直接卡在0不动了

[MengLi-l1]

这个你得自己排查下环境和第三方库的问题了，代码应该是可以运行的，我自己测试过，你试下三个模块单独训练，排查下问题。

[ZXX-Andrew]

这个你得自己排查下环境和第三方库的问题了，代码应该是可以运行的，我自己测试过，你试下三个模块单独训练，排查下问题。

嗯嗯，我自己再看看吧，感谢关注和回复，真的非常感谢！!

[ningyuv]

我就是尝试了一下RHSEDB数据集，用你的代码跑了好多次，最高才82%。是有什么特殊技巧吗？

我跑了同样是82%。在RHSEDB上，SDNet指标和论文中一致，SegNet和ExtractNet的train IOU和论文中一致，SegNet的test IOU 0.77，ExtractNet的test IOU 0.82