# TensorFlow2.0_InceptionV3

**Repository Path**: futureflsl/TensorFlow2.0_InceptionV3

## Basic Information

- **Project Name**: TensorFlow2.0_InceptionV3
- **Description**: A TensorFlow_2.0 implementation of InceptionV3.
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2020-07-09
- **Last Updated**: 2020-12-20

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# TensorFlow2.0_InceptionV3
A TensorFlow_2.0 implementation of InceptionV3.

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

## Train
1. Requirements:
+ Python >= 3.6
+ Tensorflow == 2.0.0-rc2
2. To train the InceptionV3 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
```
3. Run the script **split_dataset.py** to split the raw dataset into train set, valid set and test set. The dataset directory will be like this:
 ```
|——dataset
   |——train
        |——class_name_1
        |——class_name_2
        ......
        |——class_name_n
   |——valid
        |——class_name_1
        |——class_name_2
        ......
        |——class_name_n
   |—-test
        |——class_name_1
        |——class_name_2
        ......
        |——class_name_n
```
4. Change the corresponding parameters in **config.py**.
5. 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:
+ [ResNet34, ResNet50, ResNet101, ResNet152](https://github.com/calmisential/TensorFlow2.0_ResNet)
+ [InceptionV3](https://github.com/calmisential/TensorFlow2.0_InceptionV3)

## References
1. The original paper :https://arxiv.org/abs/1512.00567
2. Google official implementation of InceptionV3 (TensorFlow 1.x): https://github.com/tensorflow/models/blob/master/research/slim/nets/inception_v3.py
3. https://www.jianshu.com/p/3bbf0675cfce
4. Official PyTorch implementation of InceptionV3 : https://github.com/pytorch/vision/blob/master/torchvision/models/inception.py