search

huang229 / auto_tooth_arrangement

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

Pre-trained model #8

Open mrcabellom opened 9 months ago

mrcabellom commented 9 months ago

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

huang229 commented 9 months 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.

huang229 commented 9 months 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 3 months ago

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