angryhen / model_tensorflow2.0

learning tensorflow2.0

model tensorflow2.x

About the tutorial

---

Can refer to the following URL： https://www.jianshu.com/u/9f7146c71955

base_model

---

Various basic models code by tensorflow2.x,now included:

• VGG16,
• ResNet50
• MobileNet_v2
• SE-Resnet

Update log

• TF-record
• Config-file
• Train
• Eval
• Predict
• Performance

Prepare Datasets

1. use split_dataset.py to split train/valid/test data

# change path of dataset_dir in config.py's split_dataset
python split_datdaset.py 

2. create TF-record

python create_tfrecord.py

Change param

All parameters are saved in config.py , and you can modify it to suit yout training.

Train

python train.py

Eval

• TODO