xavysp / TEED

TEED: Tiny and Efficient Edge Detector
MIT License
179 stars 17 forks source link

Only detect document edges #22

Open UserJTZhang opened 4 months ago

UserJTZhang commented 4 months ago

Thank you very much for your algorithm. I am currently using it for document edge detection. In this process, I do not want to detect the edges of document content (text or image content), nor do I want to detect the edges of the background, only keeping the edges of the document. But obviously, the results I obtained contain too many other edges, and I see that the feature maps given by out1 and out2 also contain many other edges. As shown in the figure, I have converged after 24 epochs of training, with approximately 50000 pairs of image data. I hope to receive your results reply and provide some directional suggestions. Thank you very much.

UserJTZhang commented 4 months ago

image

xavysp commented 3 months ago

Thank you very much for your algorithm. I am currently using it for document edge detection. In this process, I do not want to detect the edges of document content (text or image content), nor do I want to detect the edges of the background, only keeping the edges of the document. But obviously, the results I obtained contain too many other edges, and I see that the feature maps given by out1 and out2 also contain many other edges. As shown in the figure, I have converged after 24 epochs of training, with approximately 50000 pairs of image data. I hope to receive your results reply and provide some directional suggestions. Thank you very much.

Hi, sorry for my late response: I think you need more samples, you just have 1747 iterations for an epoch. Mine are around 56k per epoch. As this is like a segmentation you should you some preprocessing or post processing in this task.

KeyWu commented 2 months ago

Thank you very much for your algorithm. I am currently using it for document edge detection. In this process, I do not want to detect the edges of document content (text or image content), nor do I want to detect the edges of the background, only keeping the edges of the document. But obviously, the results I obtained contain too many other edges, and I see that the feature maps given by out1 and out2 also contain many other edges. As shown in the figure, I have converged after 24 epochs of training, with approximately 50000 pairs of image data. I hope to receive your results reply and provide some directional suggestions. Thank you very much.

您好,想请问您一下我在执行python main.py --choose_test_data=-1时报错,您有时间能回复一下吗?我将不胜感激! Snipaste_2024-09-21_16-01-39