CNN – Apple Classification

I have the following problem statement in which I only need to predict whether a given image is an apple or not. For training only 8 images are provided with the following details:

  1. apple_1 image – 2400×1889 PNG
  2. apple_2 image – 641×618 PNG
  3. apple_3 image – 1000×1001 PNG
  4. apple_4 image – 500×500 PNG contains a sticker on top of fruit
  5. apple_5 image – 2400×1889 PNG
  6. apple_6 image – 1000×1000 PNG
  7. apple_7 image – 253×199 JPG
  8. apple_8 image – 253×199 JPG

I am thinking about using Transfer learning: either VGG or ResNet-18/34/50. Maybe ResNet is an overkill for this problem statement? How do I deal with such varying image sizes and of different file extensions (PNG, JPG)?

Any online code tutorial will be helpful. I found this example code online.


submitted by /u/grid_world
[visit reddit] [comments]

Leave a Reply

Your email address will not be published. Required fields are marked *