TexasInstruments / edgeai-yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite. Forked from https://ultralytics.com/yolov5
https://github.com/TexasInstruments/edgeai
GNU General Public License v3.0
644 stars 120 forks source link

For the specified number of key points custom dataset debug(针对自定义关键点数量的yolopose的debug) #102

Open divided7 opened 1 year ago

divided7 commented 1 year ago

我总结了一下需要改的地方: https://zhuanlan.zhihu.com/p/603799078 如果对您有帮助给个点赞收藏哈哈哈

nomaad42 commented 1 year ago

@divided7

Thank you very much for sharing the article!

Have you encountered problem, when your custom model does not converge? If yes, how did you solve it?

I already tried to modify sigmas and loss function, it helped a bit, but not significantly

divided7 commented 1 year ago

@divided7

Thank you very much for sharing the article!

Have you encountered problem, when your custom model does not converge? If yes, how did you solve it?

I already tried to modify sigmas and loss function, it helped a bit, but not significantly

I also meet some bug in modify sigmas. In a word, you can print the component of each lkpt, if every loss is going down, the code is right

nomaad42 commented 1 year ago

Let me check it! Thank you!