xdbai-source / RPNet

0 stars 0 forks source link

RPNet

Source code for Our RPNet:

  1. Our code is based on PyTorch deep learning framework. Please install and configure PyTorch operating environment on your computer.

  2. Download all the codes to your local computer.

  3. Download URC test dataset by the link: https://pan.baidu.com/s/1driZAj_K6J2elR8RXO-9zA?pwd=wra8 code:wra8

    Download checkpoint by the link: https://pan.baidu.com/s/1y4A5usEFuWtBVoHiDtTuRA?pwd=n6m5 code:n6m5

  4. Unrar the above two files to the codes folder.

  5. Run 'eval.py' to test the counting performance of RiceNet

  6. File 'train.py' provides the code for network training. Hope it will have reference value in your crop counting task.

  7. File 'models.py' gives the code for the network pipeline of RPNet.