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
  1. Add torchvision_classif.py to models. This file integrated ResNet, DenseNet, Vision Transformer, Swin Transformer, and Swin Transformer V2 based on the Torchvision library.
  2. Delete the previous resnet.py