55 lines
1.7 KiB
Markdown
55 lines
1.7 KiB
Markdown
# Sumasen Trainer
|
|
---
|
|
### Train and create a new Model
|
|
|
|
1) Select New from left menu
|
|

|
|
|
|
2) Enter image folder where images are saved in sub folder as their category
|
|
|
|
**This will assign a outpu folder to arrage images, you can modify if required**
|
|

|
|
|
|
Eg:
|
|
`image_classifier/dataset/raw_data`
|
|

|
|
|
|
|
|
3) Select input shape
|
|
This will be used to resize image into
|
|

|
|
|
|
4) Select Batch Size
|
|
This is the number of image used to train as a batch
|
|

|
|
|
|
5) Click arrange images button
|
|
This will rearrange the images seected above and arrange them as train, test and validation sets
|
|

|
|
|
|
6) a) Enter a name for model, the model will be saved under previouly selected output folder
|
|
b) also select number of epochs the training should run
|
|
)
|
|
|
|
7) training progrss
|
|

|
|
|
|
8) Training completed.
|
|
The model will be saved in the output folder
|
|

|
|
|
|
|
|
### Predicting a new image
|
|
|
|
1) Select "Existing model" from left pane
|
|
|
|

|
|
|
|
2) Enter a folder to list all models available (if a training session is done previously the output folder will be there automatically)
|
|
3) Click load model and select a model available
|
|

|
|
|
|
4) On the right pan select images shape that used for this model and a new image to predict,
|
|

|
|
|