Skip to content

Latest commit

 

History

33 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TRAINING AND TESTING DATASET

  • python code for making the .pth CNN model (EfficientNet B0 by default)

Steps

  1. Download the image dataset refer here
  2. Split the image dataset refer here
  3. Clone this repository and check the changeme at train_bruises.py and train_ripeness.py
  4. Run the command in CLI
  • for the bruises
python train_bruises.py
  • for the ripeness
python train_ripeness.py

Other Remarks

  • output graphs are in .png
  • output CLIs are in .txt
  • variable epoch=1 is default at main()
  • img dataset path is set to be the same directory as the .py as seen on the .gitignore

Timestamp for AlexNet

this took a while

About

training and testing dataset

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages