PranayPalem
/

BYOL_Mammogram

Image Classification

medical-imaging

self-supervised-learning

computer-vision

Model card Files Files and versions

1.11 GB

1 contributor

History: 3 commits

This model has 1 file scanned as unsafe.

PranayPalem's picture

📖 Add example usage script

853f08d 2 months ago

.gitattributes

1.52 kB

initial commit 2 months ago
CLASSIFICATION_GUIDE.md

9.38 kB

🏥 Add BYOL Mammogram Classification Model 2 months ago
README.md

6.48 kB

🏥 Add BYOL Mammogram Classification Model 2 months ago
classification_config.json

251 Bytes

🏥 Add BYOL Mammogram Classification Model 2 months ago
example_usage.py

4.81 kB

📖 Add example usage script 2 months ago
inference_classification.py

9.69 kB

🏥 Add BYOL Mammogram Classification Model 2 months ago
mammogram_byol_best.pth
Detected Pickle imports (4)
- "torch.FloatStorage",
- "torch.LongStorage",
- "collections.OrderedDict",
- "torch._utils._rebuild_tensor_v2"
What is a pickle import?
553 MB
xet

🏥 Add BYOL Mammogram Classification Model 2 months ago
mammogram_byol_final.pth
Detected Pickle imports (9)
- "__builtin__.getattr",
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage",
- "wandb.sdk.wandb_settings.Settings",
- "torch.LongStorage",
- "collections.OrderedDict",
- "__builtin__.set",
- "wandb.sdk.wandb_run.Run",
- "wandb.sdk.wandb_config.Config"
How to fix it?
553 MB
xet

🏥 Add BYOL Mammogram Classification Model 2 months ago
requirements.txt

171 Bytes

🏥 Add BYOL Mammogram Classification Model 2 months ago
train_byol_mammo.py

33.7 kB

🏥 Add BYOL Mammogram Classification Model 2 months ago
train_classification.py

18.5 kB

🏥 Add BYOL Mammogram Classification Model 2 months ago