/WheatNet

Primary LanguagePython

WheatNet

project for WheatNet

This project is run on the Pytorch 0.3 version(it will be updated in to 0.4 version later)

The hardware requirement is 2 GPU with at least 12 GB on each one

When you want to use the code to train a model so as to predict the percentage of the maturity

  1. change the code in the Main_PCT.py

    src is the image repository dst is the result saving directory

  2. put keyfile in the 'keyfile' folder and change the keyfile.py with the keyfile names