Mhara/melanoma_classification

By Mhara · ↓ 0 · ♥ 0 · Updated 2025-04-28

What's in this repo

This page lists the files in Mhara/melanoma_classification, including best_model_domain_discriminative.onnx.

1 ONNX models (.onnx files)
71 files, including 40 .py, 6 .sh, 4 .yaml, 4 .txt, 3 .md, 3 .ipynb
Last updated 2025-04-28

ONNX files

File	Size	Download
weights/best_model_domain_discriminative.onnx	—	Download

How to use

These ONNX files can be loaded with ONNX Runtime. To inspect a model graph, upload the file to the Netron viewer. Download links point to the original files on Hugging Face.

README

Results

Installation

Please check INSTALL.md for installation instructions.

Training

See TRAINING.md for training and fine-tuning instructions.

Model info

onnx region:us

↓ 0 ♥ 0

Created: 2025-04-28

Updated: 2025-04-28

View on Hugging Face

Files (71)

.dockerignore

.gitattributes

.gitignore

.pre-commit-config.yaml

INSTALL.md

README.md

TRAINING.md

configs/best_run.yaml

configs/dino_vit_small.yaml

configs/dino_vit_small_dist.yaml

data/ISIC_2020_full.csv

docker/Dockerfile

docker/Dockerfile_cpu

environment.yml

melanoma_eval.py

melanoma_train.py

notebooks/ANOVA_and_fairness.ipynb

notebooks/skin_tone_analysis.ipynb

notebooks/skin_tone_estimation.ipynb

predict.py

pyproject.toml

requirements.txt

requirements_dev.txt

requirements_inference.txt

requirements_torch.txt

run_with_submitit.py

src/datasets/data_processing.py

src/datasets/datasets.py

src/datasets/sampler.py

src/engine/engine.py

src/engine/scheduler.py

src/evaluation/metrics.py

src/models/backbones/ConvNeXt.py

src/models/backbones/ConvNeXtV2.py

src/models/backbones/DinoV2.py

src/models/backbones/EfficientNet.py

src/models/backbones/ResNetPyramid.py

src/models/backbones/ResnetSingleScale.py

src/models/backbones/SeResNeXt.py

src/models/backbones/convnext_isotropic.py

src/models/layers/attention.py

src/models/layers/block.py

src/models/layers/dino_head.py

src/models/layers/drop_path.py

src/models/layers/layer_scale.py

src/models/layers/mlp.py

src/models/layers/patch_embed.py

src/models/layers/swiglu_ffn.py

src/models/losses/criterion.py

src/models/melanoma_classifier.py

src/models/optim_factory.py

src/models/utils/utils.py

src/scripts/estimate_ita.py

src/scripts/export_model.py

src/scripts/generate_skin_masks.py

src/scripts/get_results.py

src/scripts/log_to_csv.py

src/utils/argparser.py

src/utils/distributed.py

src/utils/logging_utils.py

src/utils/utils.py

src/visualization/visualize_images.py

test.sh

train.sh

train_config.sh

train_dino.sh

train_distributed.sh

train_kaggle.sh

weights/best_model_domain_discriminative.onnx ONNX

weights/best_model_domain_discriminative.pt

weights/best_model_domain_discriminative.pth

Related models

More from Mhara

No other models by this author.

More models

No task type declared.