limhoyeon / ToothGroupNetwork

3D Dental surface segmentation with Tooth Group Network
175 stars 41 forks source link

Training on custom data #29

Open SashaNix opened 8 months ago

SashaNix commented 8 months ago

Good day, I see in the inference branch that you use several models, like 0809_sched_v2_fixed_Flip(fixed)_weight0.1_val.h5 0809_reverse_sched_v2_fixed_Flip(fixed)_weight0.1_val 0809_reverse_sched_fixed_tf_cbl_2.0_second_Flip_val 0808_sched_fixed_tf_cbl_2.0_second_Flip_val 0805_mask_model_val

Is there any code related to training this models?

limhoyeon commented 7 months ago

Thank you for your interest in my research. Im sorry but I dont have the codes you want now.. But these models are not much different from the codes in main branch except flip augmentation, scheduler setting and loss for contrastive boundary learning.. Thank you

SashaNix commented 7 months ago

I see, but can you tell what type of models are these? in the main branch I see 3 type of models training: tgnet, tsegnet and pointnet

limhoyeon commented 7 months ago

I used my model(tgnet)!