PranayPalem
/

BYOL_Mammogram

Image Classification

medical-imaging

self-supervised-learning

computer-vision

Model card Files Files and versions

1.11 GB

Ctrl+K

Ctrl+K

1 contributor

History: 3 commits

PranayPalem's picture

📖 Add example usage script

853f08d 9 months ago

.gitattributes

1.52 kB
initial commit 9 months ago
CLASSIFICATION_GUIDE.md

9.38 kB
🏥 Add BYOL Mammogram Classification Model 9 months ago
README.md

6.48 kB
🏥 Add BYOL Mammogram Classification Model 9 months ago
classification_config.json

251 Bytes
🏥 Add BYOL Mammogram Classification Model 9 months ago
example_usage.py

4.81 kB
📖 Add example usage script 9 months ago
inference_classification.py

9.69 kB
🏥 Add BYOL Mammogram Classification Model 9 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 9 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 9 months ago
requirements.txt

171 Bytes
🏥 Add BYOL Mammogram Classification Model 9 months ago
train_byol_mammo.py

33.7 kB
🏥 Add BYOL Mammogram Classification Model 9 months ago
train_classification.py

18.5 kB
🏥 Add BYOL Mammogram Classification Model 9 months ago