Skip to content

Create a Skin Condition classification task#678

Merged
tharvik merged 20 commits intodevelopfrom
swiss-tph-demo
May 31, 2024
Merged

Create a Skin Condition classification task#678
tharvik merged 20 commits intodevelopfrom
swiss-tph-demo

Conversation

@JulienVig
Copy link
Copy Markdown
Collaborator

@JulienVig JulienVig commented May 27, 2024

Add a skin condition multi-class classification task as a predefined task in Disco for the upcoming demo. The SCIN dataset is described here.

  • original model previously used for this task currently only predicts the benign class
  • create a sample image dataset to provide to users (original dataset is 3GB and requires creating an account on Kaggle and creating the csv label mapping file) -> Available on the google cloud bucket
  • Testing a model fails with error
  • fine tune architecture and hp. I tried using a pre-trained mobilenet (a334219, commented code in skin_mnist.ts) but encountered Discrepancy between training logs and manually evaluating metrics #679 and decided to settle on a smaller handmade CNN for now.
    For 800 images
    Screenshot 2024-05-30 at 12 31 58
  • Allow connecting a folder rather than individual files -> moved to Connecting folder displays alarming "Upload" warning message #680
  • Add an example of how to create csv file to connect data -> docs/dataset_csv_creation.ipynb
  • At test time, show label names rather than integer numbers
    Before:
    Screenshot 2024-05-30 at 11 29 53
    Now:
    Screenshot 2024-05-30 at 12 33 24

@JulienVig JulienVig marked this pull request as ready for review May 30, 2024 11:04
@JulienVig JulienVig requested review from EduardDurech and tharvik May 30, 2024 11:05
Copy link
Copy Markdown
Collaborator

@tharvik tharvik left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

well done (and sorry that you had to rush it)! a bit of cleanup but nothing vital

@JulienVig JulienVig changed the title Integrate the skin disease classification task Create a Skin Condition classification task May 30, 2024
@JulienVig
Copy link
Copy Markdown
Collaborator Author

@tharvik do you want to review the latest changes or shall I merge the PR?

@tharvik tharvik merged commit 1f0f0e5 into develop May 31, 2024
@tharvik tharvik deleted the swiss-tph-demo branch May 31, 2024 08:59
@JulienVig JulienVig mentioned this pull request Jun 5, 2024
6 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants