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: 2 commits

This model has 1 file scanned as unsafe.

PranayPalem's picture

🏥 Add BYOL Mammogram Classification Model

d921913 8 months ago

.gitattributes

1.52 kB
initial commit 8 months ago
CLASSIFICATION_GUIDE.md

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

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

251 Bytes
🏥 Add BYOL Mammogram Classification Model 8 months ago
inference_classification.py

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

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

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

18.5 kB
🏥 Add BYOL Mammogram Classification Model 8 months ago