Lightweight neural network for binary classification of cancer metastasis in bone scan images using PyTorch. Code repository for the paper "A Lightweight Convolutional Neural Network for Detection of Osseous Metastasis using Feature Fusion and Attention Strategies". Presented at CVIPPR 2024.
Rationale:
Although the same in practice given 2 classes, BCELoss matches the function in the RRL Status:
Not yet implemented as BCELoss doesn't support different tensor sizes for input & target, and I have yet to find a workaround
Rationale:
Although zoom was initially planned to be included in the list of transformations, transforms v1 doesn't have a zoom transformation Status:
Awaiting PyTorch 15.0 release
Rationale:
transforms v1 doesn't allow for exclusion of 0 from range, so there's a chance an image won't be translated even if the augmentation picked for that image is translate Status:
Awaiting PyTorch 15.0 release