Data yang dipakai untuk melatih model klasifikasi kacang ini
Dataset berasal dari Kaggle dengan nama Tree Nuts Image Classification, berisi gambar berbagai jenis kacang pohon yang dikurasi khusus untuk keperluan klasifikasi gambar menggunakan deep learning.
Seluruh gambar sudah dalam format JPG berukuran 224x224 piksel dengan tiga channel warna RGB, dan sudah terbagi rapi ke dalam folder train, valid, dan test.
Untuk meningkatkan variasi data training dan mencegah overfitting, dilakukan augmentasi data secara otomatis selama proses training berlangsung:
Almonds
Brazil Nuts
Cashews
Coconut
Hazelnuts
Macadamia
Pecans
Pine Nuts
Pistachios
Walnuts
Jumlah data pada dataset ini tergolong terbatas, kurang dari 1500 gambar untuk seluruh kelas. Karena itu, model tidak dilatih dari nol, melainkan menggunakan teknik transfer learning dengan VGG16 yang sudah dilatih sebelumnya pada jutaan gambar di ImageNet, lalu disempurnakan dengan fine-tuning agar lebih akurat mengenali fitur khas tiap jenis kacang.