Source code for Our RiceNet:
Our code is based on PyTorch deep learning framework. Please install and configure PyTorch operating environment on your computer.
Download all the codes to your local computer.
Download URC dataset by the link: https://pan.baidu.com/s/1L2g6HMjjY7kaJmGn9HTNzA?pwd=g1rq code:g1rq
Download checkpoint by the link: https://pan.baidu.com/s/1o-X2wrvZTTmfxGbfWfUfsQ?pwd=1w3y code:1w3y
Unrar the above two files to the codes folder.
Run 'eval.py' to test the counting performance of RiceNet
Run 'detect.py' to test the sizing and locating of RiceNet
File 'train.py' provides the code for network training. Hope it will have reference value in your crop counting task.
File 'models.py' gives the code for the network pipeline of RiceNet.