Connect data loader and so we can use the Covid dataset - Githubissues

Lightning-Universe / lightning-Covid19

Classification for covid-19 chest X-ray images using Lightning

https://pytorchlightning.github.io/lightning-Covid19

MIT License

56 stars 18 forks source link

Connect data loader and so we can use the Covid dataset #2

Closed oplatek closed 4 years ago

oplatek commented 4 years ago

Replace the Cocodataset in the EfficientDet.Pytorch by Covid dataset

See the training script for models which we want to use https://github.com/toandaominh1997/EfficientDet.Pytorch/blob/97f2851b6063da59a7ae73dadfc4b5e5022473e3/train.py#L182

See the Covid dataset dataloader https://github.com/mlmed/torchxrayvision/blob/master/torchxrayvision/datasets.py#L814