yeyuxmf / auto_tooth_arrangement

Tooth arrangement, Medical orthodontics,Neural networks, Deep learning, Transformer, Pytorch, Python
35 stars 9 forks source link

Pre-trained model #8

Open mrcabellom opened 1 year ago

mrcabellom commented 1 year ago

Could you enable again the link of the pre-trained model?

yeyuxmf commented 1 year ago

@mrcabellom Sorry, I only saw a reply from you now. The existing model was mistakenly deleted on the online drive, and I don't have it either. As long as the environment is installed and the optimizer is changed to Adam, this code can be directly trained using the data I provide. If you feel that the final loss value is not small enough, increase the number of epochs, and the loss value will be further reduced.

yeyuxmf commented 1 year ago

If you want to train your data, you need to change the data preprocessing code and calculate your rotation and translation matrices.

smoothumut commented 6 months ago

Hi @huang229 thanks for the great work, the model's link looks broken again.