search

nyukat / breast_cancer_classifier

Deep Neural Networks Improve Radiologists' Performance in Breast Cancer Screening
https://ieeexplore.ieee.org/document/8861376
GNU Affero General Public License v3.0
835 stars 265 forks source link

ImageNet weights #50

Closed githubFrandeveloperb354 closed 2 years ago

githubFrandeveloperb354 commented 2 years ago

is it possible to share the weights of the pre-trained model on imagenet?

wooginawunan commented 2 years ago

Hi, for the patch classifier, we use DenseNet built in PyTorch. Please refer to https://pytorch.org/vision/0.8/_modules/torchvision/models/densenet.html for details about the pre-trained model on imagenet.