无法训练自己的数据
wangduyang opened this issue · 2 comments
wangduyang commented
训练自己的数据的时候，修改了joint_nums 依然无法训练，很好奇这是为啥。
提示：“the number of joints should be 22” （我的数据是21个关键点）
在debug的时候，CocoKeypoints.py 中的 joints_list 一会18 一会22。
zranguai commented
在自定义数据集上，RandomHorizontalFlip的时候shape会在（1， 22， 3）和（1， 18， 3）之间一直变。
wusaisa commented
您好，我也想训练自己的数据，关键点数量和输入分辨率都有改变，请问除了在default.py里修改num_joints之外，还需要修改哪些代码吗？谢谢~