CottonWeeds
Deep Learning Models for cotton&weeds classification.
requirements
- pytorch
- torchsummary
- tensorboard
- PIL
- Scikit-learn
Dataset Preparing and Downloading
-
To prepare your own dataset, you can run
-
You can also have a try on our open-sourced dataset at:
Usage
- To train the models, just specify the name of the models, and then run
python train.py. - To test the images, just specify the name of the models, and then run
python test.py. - To eval new data, just specify the name of the models, and then run
python eval.py.