Qi Han
Qi Han
If you want to train DHT on your custom dataset, the straight line annotations is needed to calculate the parameteric space lable. Then train your model with the parameteric space...
I think this may caused by the incompatible cuda version, you can try to compile it using cuda10.
文件目录和名称是否按照示例样式给出?
由于问题我难以复现,能否把你的测试图文件夹部分发给我邮箱,我来debug一下呢? [email protected]
@1021151558 你的目录设置是没有问题的,问题可能发生在,dataloader返回的图像是0-255的uint8格式,而你的修改将其转换为了0-1的float32格式,导致后面f-measure在根据阈值计算时发生错误. 建议修改正确的dataloader返回格式,以支持对tiff图像的读取并返回正确的格式.
感谢关注与提醒，应该是predictions和gt都是png格式的。 已经在Readme中更正example，谢谢。
dht借助于deep cnn的feature表达，其原始语义与hough空间的不一定相同，属于预期现象，因为是端到端训练，模型会优化中间空间的表达以更适应当前任务。 cuda反传代码可以自己通过导函数的定义来验证。