| from train import Train |
| from test import Test |
| from configuration import DatasetName, ModelArch |
| from pca_utility import PCAUtility |
| if __name__ == '__main__': |
| '''use the pretrained model''' |
| tester = Test() |
| tester.test_model(ds_name=DatasetName.w300, |
| pretrained_model_path='./pre_trained_models/ASMNet/ASM_loss/ASMNet_300W_ASMLoss.h5') |
|
|
| '''training model from scratch''' |
| |
| |
| pca_calc = PCAUtility() |
| pca_calc.create_pca_from_npy(dataset_name=DatasetName.w300, |
| labels_npy_path='./data/w300/normalized_labels/', |
| pca_percentages=90) |
|
|
| |
| trainer = Train(arch=ModelArch.ASMNet, |
| dataset_name=DatasetName.w300, |
| save_path='./', |
| asm_accuracy=90) |
|
|
