YigitTurali / HydraViT

HydraViT is a PyTorch implementation of the HydraViT model, an adaptive multi-branch transformer for multi-label disease classification from chest X-ray images. The repository provides the necessary code to train and evaluate the HydraViT model on the NIH Chest X-ray dataset.

https://arxiv.org/abs/2310.06143

Geek Repo

Github PK Tool