search

huang229 / auto_tooth_arrangement

Tooth arrangement, Medical orthodontics，Neural networks, Deep learning, Transformer, Pytorch, Python
32 stars 8 forks source link

你好 #12

Open Fallda1 opened 4 months ago

Fallda1 commented 4 months ago

hello, 请问可以加个微信吗? 对牙齿重排这块的实现比较感兴趣,目前想通过计算LDA来实现. 我的微信号:zxhsqzhx. 还有看到您说改下优化器可以直接训练,在load_train_data.py 里的tooth rotate over似乎一直有个bug说operands could not be broadcast together with shape (1,) (14,), 请问您知道怎么解决吗?

huang229 commented 4 months ago

代码是毫无bug的，只是环境的配置在ubuntu下更容易才成功，因为pytorch3D对windows支持很差，如果是深度学习新手，建议ubuntu下配置环境。微信已添加