jkbdnj / jakub-dunaj-bachelors-thesis

Bachelor's thesis project on Plant Disease Classification for Vienna university of Technology. Includes LaTeX document and accompanying web application.
0 stars 0 forks source link

Create a script training a CNN using the created dataset #2

Closed jkbdnj closed 2 days ago

jkbdnj commented 2 weeks ago

Task description

Create a script that would use to train a CNN to classify diseases from plant leaf images. The important things to consider are:

jkbdnj commented 2 weeks ago

NOTE: When selecting a CNN architecture, I am also limited by the architectures provided by keras that are pre-trained on Imagenet dataset: https://keras.io/api/applications/.

The most promising CNN model is the: EfficientNetB0.

jkbdnj commented 2 weeks ago

NOTE: Typical transfer learning workflow + fine-tuning: https://keras.io/guides/transfer_learning/#the-typical-transferlearning-workflow.

There is also no need to resize the images in the datset because they have already been resized oto 256 256 pixels. The EfficientNetB0 has the spatial input dimensionality of 256 256 pixels.