Skip to content

Latest commit

 

History

History
60 lines (42 loc) · 919 Bytes

README.md

File metadata and controls

60 lines (42 loc) · 919 Bytes
Raw

CNN-Melanoma-Detector

Requirements

  • python 3.10.13

Create environment

python -m venv venv

Start the environment

# linux
source venv/bin/activate

# windows
venv\Scripts\activate

Download dataset

Download Kaggle melanoma dataset here
Extract the main folder (melanoma_cancer_dataset folder).

Train

python train_model.py <EPOCHS_NUMBER>

Run API

fastapi dev main.py

Endpoints

POST

body_type: form

params: (image: file)

NOTE: use images sizes around 224x224 to better accuracy, the endpoint rescales the image input to 224x224 but that can cause some precision issues.

http://127.0.0.1:8000/predict

GET

body_type: no need it

params: no need it

http://127.0.0.1:8000/accuracy