xavysp / TEED

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

The setting problem of TEED training parameters #8

Open KinberAlan opened 11 months ago

KinberAlan commented 11 months ago

Hello, sir. When I use your TEED model for training, I get the.pth file. However, the quality of the prediction results obtained by using the weight file I trained myself is very poor, while using the weight file in your source code can obtain good prediction results. I don 't know if there is a problem with the parameter setting. Moreover, I used the BIPED dataset to train 8 epochs in less than 5 minutes, while the training time mentioned in your article is 30 minutes. I suspect that something must have gone wrong. Can you tell me your training parameter settings ? The first image is the prediction result obtained by using the weight file obtained by my own training, and the second is the prediction result obtained by using the weight file in your source code. Hope to receive your reply, thank you very much ! 1 1

xavysp commented 11 months ago

Hi, thanks for using TEED. looking at the issue 7, I think you fund the problem didn't you?

P. D. What an impressive result.

KinberAlan commented 11 months ago

Hi, thanks for using TEED. looking at the issue 7, I think you fund the problem didn't you?

P. D. What an impressive result.

Yes,!Thank you very much for your attention and answers! Wish you a good day. I will expand my data set to retrain the model.