Skip to content

Training the model with custom data #28

Answered by Thomasbehan
upstream-team asked this question in Q&A
Discussion options

You must be logged in to vote

Absolutely, you can extend the capability of SkinvestigatorAI to detect and classify skin problems like rashes and acne. Just add your images to the /data directory (if there isn't one you can create one). Organize them in test, train, and validation subfolders. Within each of these subfolders, create a separate folder for each diagnosis you want the AI to detect. For instance: /data/validation/melanoma/image.png. or /data/train/eczema/image.jpg
You can store as many images in each diagnoses folder as you like.

Once your data is organized, you can train the AI by running the training script: /skinvestigatorai/core/ai/train.py.

Hope this helps.

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by Thomasbehan
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
enhancement New feature or request question Further information is requested
2 participants
Converted from issue

This discussion was converted from issue #27 on September 16, 2023 21:17.