TensorFlow2.0_ResNet

A ResNet(ResNet18, ResNet34, ResNet50, ResNet101, ResNet152) implementation using TensorFlow-2.0

See https://github.com/calmisential/Basic_CNNs_TensorFlow2.0 for more CNNs.

Train

Requirements:

Python >= 3.6
Tensorflow == 2.0.0

To train the ResNet on your own dataset, you can put the dataset under the folder original dataset, and the directory should look like this:

|——original dataset
   |——class_name_0
   |——class_name_1
   |——class_name_2
   |——class_name_3

Run the script split_dataset.py to split the raw dataset into train set, valid set and test set.
Change the corresponding parameters in config.py.
Run train.py to start training.

Evaluate

Run evaluate.py to evaluate the model's performance on the test dataset.

The networks I have implemented with tensorflow2.0:

References

The original paper: https://arxiv.org/abs/1512.03385
The TensorFlow official tutorials: https://tensorflow.google.cn/beta/tutorials/quickstart/advanced