DenseNet in fastai?

Part 1 (2018)
1

Hi!

Wanted to know if there is DenseNet implementation in fastai library. After reading this paper i want to test DenseNet on NIH Chest X-ray Dataset. Seems that it is good option, as I get only around 70% accuracy with binary classification (Pathology/No pathology) with Resnet variations.

2 Likes
2

Full support for DenseNet seems to have been dropped in version 1.0 of fastai. However it is easy enough to use the Pytorch model directly.

from torchvision.models import densenet121

def dn121(pre): return children(densenet121(pre))[0]

learn = create_cnn(data, dn121)
2 Likes