TensorFlow2.0_Image_Classification
TensorFlow2.0_Image_Classification copied to clipboard
A TensorFlow_2.0 implementation of AlexNet and VGGNet.
1. Added 25 models from Keras API 2. Auto Evaluation + Testing after training 3. Added more metrics (AUC, Precision, Recall) 4. Added early stop (patient=5) callback 5. Added Saving...
Sometimes I thought only total acc would not satisfy my request, so is it possible to show each class test acc during the evaluation?
Hi, I see that the prepare_data.py loads data from directory, so is it limited to single classification task? if not, how to modify it in order to use it in...