Closed LittlePeng closed 7 years ago
@LittlePeng Check if there is any broken image or irrelevant file(i.e., *.jpg) in your subfolders.
@ugtony thanks
for some mtnn aligned .png files has a 90 degree,then i edit (rotation) in mac. after clean those files, it works.
command:
➜ facenet git:(master) ✗ python src/facenet_train_classifier.py --logs_base_dir ./logs/ --models_base_dir ./model/ --data_dir ~/data2/mscelebv1/merged_mtcnn --image_size 160 --model_def models.inception_resnet_v1 --lfw_dir ~/data2/mscelebv1/merged_mtcnn_lfw2 --lfw_pairs ~/data2/mscelebv1/merged_mtcnn_lfw/pairs.txt --optimizer RMSPROP --learning_rate -1 --max_nrof_epochs 80 --keep_probability 0.8 --random_crop --random_flip --learning_rate_schedule_file data/learning_rate_schedule_classifier_casia.txt --weight_decay 5e-5 --center_loss_factor 1e-4 --center_loss_alfa 0.9 --epoch_size 2 --batch_size 50
output:
my dataset, 5K+ classes 30w images: Total number of classes: 5285 Total number of examples: 301320
reference to #105 OutOfRangeError occurred during training
i try set small batch_size,
--epoch_size 2 --batch_size 50
, but crash at 2nd epoch ?