CNN_Own_Dataset icon indicating copy to clipboard operation
CNN_Own_Dataset copied to clipboard

CNN example for training your own datasets.

CNN Own Dataset

Try your data instead of MNIST data in CNN tutorials.

Before run the source code

You must prepare the dataset like following.

The root directory of your own dataset.

  • Class 1
    • image 1-1
    • image 1-2 ...
    • image 1-n
  • Class 2
    • image 2-1
    • image 2-2
      ...
    • image 2-n
  • Class 3
    ...
  • Class N

How to run?

Enter the commend on your terminal like following.

$ python run.py

...

Enter the path where the images are located.
>> {Enter the path where root directory of your own dataset.}

And following commend will help you when you run the source code.

$ python run.py -h

First commit: May.24.2017 Developed by YeongHyeon Park (Korea)