-
About
We can train custom dataset using this program with pytorch.
-
File descripttion
file description ./custom_traindata/* training image folds. (fold's names are label) ./custom_testdata/* testing image folds. (fold's names are label) ./exampleData example data for test excuting. -
Requirements
python 3.x torch torchvision jupyter -
How to run
You can use your own image datasets.
Copy your image dataset folders(folder name is image's label) to ''./my_traindata/'' and ''./my_testdata/''
running example:
cd ~/my_classifier cd ./exampleData/ mv my_traindata/ .. mv my_testdata/ .. jupyter notebook run the 'my_Classifier.ipynb' in your jupyter environment.output:
kmk324/My_Classifier
Folders and files
| Name | Name | Last commit date | ||
|---|---|---|---|---|
