Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Skin Cancer Detection | 1. Dataset Prep(augmentation, Noise Removal, etc) #129

Closed
1 task done
Sindhu-2004 opened this issue May 23, 2024 · 11 comments
Closed
1 task done

Comments

@Sindhu-2004
Copy link

Is your feature request related to a problem? Please describe.

A skin cancer detection model using TensorFlow to classify images as malignant or benign.

Describe the solution you'd like

Dataset :
[Skin Cancer Dataset] https://www.kaggle.com/datasets/nodoubttome/skin-cancer9-classesisic

Approach : In this skin cancer detection project we implement a convolutional neural network (CNN) with the EfficientNetB7 architecture, leveraging transfer learning for improved performance. By loading a dataset of skin images labeled as malignant or benign, using Pandas and Numpy for data handling and Matplotlib for visualization. The dataset is split into training and validation sets, with images preprocessed by resizing and normalizing pixel values. EfficientNetB7, pre-trained on ImageNet, serves as the backbone for feature extraction, with its layers frozen to retain learned weights. The model is built using Keras Functional API, incorporating layers for flattening, dense connections with 256 units each, ReLU activation functions for non-linearity, Batch Normalization layers for training stabilization, and a Dropout layer with a 0.3 rate to prevent overfitting. The final layer is a single neuron with a sigmoid activation function, producing a probability score for malignancy. The model is compiled using the Adam optimizer and binary cross-entropy loss, with AUC as the evaluation metric. Training is conducted over multiple epochs, followed by performance evaluation through plots of training and validation loss and AUC metrics, demonstrating the model's effectiveness in distinguishing between malignant and benign skin lesions. This approach showcases the utility of transfer learning and advanced CNN architectures in medical image classification tasks.

Describe alternatives you've considered

No response

Additional context

No response

Code of Conduct

  • I agree to follow this project's Code of Conduct
Copy link

Congratulations, @Sindhu-2004! 🎉 Thank you for creating your issue. Your contribution is greatly appreciated and we look forward to working with you to resolve the issue. Keep up the great work!

We will promptly review your changes and offer feedback. Keep up the excellent work! Kindly remember to check our contributing guidelines

@saikrishna823
Copy link

I would like to contribute to this issue.I have experience of working with Machine Learning and deep learning.I will use cnn,and other pretrained models for prediction.I am GSSOC 2024 participant.

@SrijanShovit
Copy link
Owner

@Sindhu-2004 proceed with dataset prep first;

@SrijanShovit SrijanShovit changed the title Skin Cancer Detection Skin Cancer Detection | 1. Dataset Prep(augmentation, Noise Removal, etc) May 24, 2024
@SrijanShovit
Copy link
Owner

I find your approach just copied from some AI stuff. If yes, don't repeat it again.

@saikrishna823
Copy link

@SrijanShovit ,can you assign any issues to me?

@a-di-ti
Copy link

a-di-ti commented May 31, 2024

Hi, looking forward to being assigned this issue for further contribution!

@SrijanShovit
Copy link
Owner

@saikrishna823 Your approach?

@saikrishna823
Copy link

First I will construct general CNN architecture then I will use pretrained models like VGG,Resnet and will also explore other pertained models for better accuracy.

@SrijanShovit
Copy link
Owner

How can you make sure all your models are less than 5 MB? Any plans for that?

@Taranpreet10451
Copy link

Can you please assign this to me.

Copy link

This issue has been automatically closed because it has been inactive for more than 7 days. If you believe this is still relevant, feel free to reopen it or create a new one. Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

5 participants