issues
search
sunlabuiuc
/
PyHealth
A Deep Learning Python Toolkit for Healthcare Applications.
https://pyhealth.readthedocs.io
MIT License
956
stars
207
forks
source link
Add models for image classification task
#175
Closed
wuzijian1997
closed
1 year ago
wuzijian1997
commented
1 year ago
Add torchvision_classif.py to models. This file integrated ResNet, DenseNet, Vision Transformer, Swin Transformer, and Swin Transformer V2 based on the Torchvision library.
Delete the previous resnet.py