susundingkai / CenterlineExtraction

2 stars 0 forks source link

CPU实现 #1

Open suetme opened 1 month ago

suetme commented 1 month ago

作者你好,请问这个算法有CPU的实现版本吗,或者有没有其他CPU类型的算法达到这个算法的效果?

susundingkai commented 1 month ago

可以用pytorch实现算子在cpu上跑。不过在cpu上用一般的中心线提取的库就能实现吧,这个算法在cpu上并没有太多优势。

suetme commented 1 month ago

确实,昨天我试了Zhang Suen细化算法对比一下效果差不多。另外我想请教一下两个问题:

  1. 图像块填充的参数(https://github.com/susundingkai/CenterlineExtraction/blob/65f5daca79d24e83780f771df490a8c557b27ed4/LibThinning2D.cu#L59-L68)该怎么根据图像大小设置?我用默认参数测试测1284*960这个像素级别的图像没问题,但是测2446*2236的图像就不行,好像是内存越界,应该是分块的问题
  2. 图像细化算法能够实现亚像素级别的中心线提取吗?
susundingkai commented 1 month ago
  1. 可以试一试图像padding几个像素之后再输入
  2. 不是很了解具体应用