| from configuration import DatasetName, D300wConf, InputDataSize, CofwConf, WflwConf | |
| from cnn_model import CNNModel | |
| from pca_utility import PCAUtility | |
| from image_utility import ImageUtility | |
| from student_train import StudentTrainer | |
| from test import Test | |
| from teacher_trainer import TeacherTrainer | |
| if __name__ == '__main__': | |
| '''test models''' | |
| '''train Teacher Networks''' | |
| trainer = TeacherTrainer(dataset_name=DatasetName.w300) | |
| trainer.train(arch='efficientNet',weight_path=None) | |
| '''Training Student Network''' | |
| '''300W''' | |
| st_trainer = StudentTrainer(dataset_name=DatasetName.w300, use_augmneted=True) | |
| st_trainer.train(arch_student='mobileNetV2', weight_path_student=None, | |
| loss_weight_student=2.0, | |
| arch_tough_teacher='efficientNet', weight_path_tough_teacher='./models/teachers/ds_300w_ef_tou.h5', | |
| loss_weight_tough_teacher=1, | |
| arch_tol_teacher='efficientNet', weight_path_tol_teacher='./models/teachers/ds_300w_ef_tol.h5', | |
| loss_weight_tol_teacher=1) | |
| '''COFW''' | |
| st_trainer = StudentTrainer(dataset_name=DatasetName.cofw, use_augmneted=True) | |
| st_trainer.train(arch_student='mobileNetV2', weight_path_student=None, | |
| loss_weight_student=2.0, | |
| arch_tough_teacher='efficientNet', weight_path_tough_teacher='./models/teachers/ds_cofw_ef_tou.h5', | |
| loss_weight_tough_teacher=1, | |
| arch_tol_teacher='efficientNet', weight_path_tol_teacher='./models/teachers/ds_cofw_ef_tol.h5', | |
| loss_weight_tol_teacher=1) | |
| '''WFLW''' | |
| st_trainer = StudentTrainer(dataset_name=DatasetName.wflw, use_augmneted=True) | |
| st_trainer.train(arch_student='mobileNetV2', weight_path_student=None, | |
| loss_weight_student=2.0, | |
| arch_tough_teacher='efficientNet', weight_path_tough_teacher='./models/teachers/ds_wflw_ef_tou.h5', | |
| loss_weight_tough_teacher=1, | |
| arch_tol_teacher='efficientNet', weight_path_tol_teacher='./models/teachers/ds_wflw_ef_tol.h5', | |
| loss_weight_tol_teacher=1) |