为什么我使用你训练好的模型得到的测试结果很差，下面是我得到的预测图

xavysp / LDC

LDC: Lightweight Dense CNN for Edge Detection

128 stars 15 forks source link

Open yx-yyds opened 6 months ago

yx-yyds commented 6 months ago

RGB_042

icyfox168168 commented 3 months ago

训练好的模型，哪里下载？没看到，能否给个地址

yx-yyds commented 3 months ago

checkpoints文件夹下

icyfox168168 commented 3 months ago

https://github.com/hpc203/LDC-onnxrun-cpp-py 我使用这个作者转换好的onnx，识别效果还可以，就是有个问题，识别的边缘太粗了，比如我识别水果，应该很细

yx-yyds commented 3 months ago

能把你预测结果发给我看看吗

icyfox168168 commented 3 months ago

a123321

icyfox168168 commented 3 months ago

像这样，本来是个水果，但是线条很粗，代码都是 https://github.com/hpc203/LDC-onnxrun-cpp-py 这个，没见哪里可以调整

icyfox168168 commented 3 months ago

a123324

yx-yyds commented 3 months ago

基于卷积神经网络得到的边缘一般都比较粗，需要用非极大值抑制才能得到更细的边缘，如果你要更细的边缘的话你可以看看用用扩散模型来进行边缘检测那篇文章，他得到的边缘很细

icyfox168168 commented 3 months ago

给文章的地址看看

yx-yyds commented 3 months ago

ZhouJin0854 commented 3 months ago

Hello brother, have you solved this problem with the thick lines?