search

hosamsherif / LiteSRGAN-and-LiteUNet

Official implementation of our paper titled "Lite-SRGAN and Lite-UNet: Towards Fast and Accurate Image Super-Resolution, Segmentation, and Localization for Plant Leaf Diseases"
MIT License
7 stars 1 forks source link

pretrained models for LiteSRGAN-and-LiteUNet #1

Open mehao opened 5 months ago

mehao commented 5 months ago

Could you provide the pretrained models for LiteSRGAN-and-LiteUNet?

hosamsherif commented 5 months ago

Hello mehao, both models LiteSRGAN and LiteUNet are trained on PlantVillage dataset, but I didn't add the checkpoints for the models here, as you can train them on any dataset you want using the instruction provided here in the repo. so you can train them on any dataset according to your task. But regarding the checkpoints for PlantVillage dataset I can add them later, because I don't have the latest checkpoints right now. but I can train them again later and provide the checkpoints for the models here.