henrhoi / BananaRecognizer

CNN for recognizing 🍌's in images

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Banana Recognizer 🍌

Processing power used:

  • GPU: 2x NVIDIA Tesla V100 32GB
  • CPU: 2x Intel Xeon Gold 6132 2.6G, 14C/28T, 10.4GT/s 2UPI, 19M Cache, Turbo, HT (140W) DDR4-2666
  • RAM: 24x 32GB = 768GB

Convolutional Neural Network: `BananaNetwork/banana_net_17_12_18.h5

  • INPUT > CONV > RELU > POOL > FC > RELU > FC
  • Build script: `CNN.py

Convolutional Netural Network: BananaNetwork/le_banana_net_26_02_19.h5

  • INPUT > CONV > RELU > POOL > CONV > RELU > POOL > FC > RELU > FC

  • Build script: CNN_LeNet.py

  • Also looking at other alternatives for design of network

File structure
β”œβ”€β”€ BananaNetwork
β”‚   β”œβ”€β”€ banana_net.h5
β”‚   β”œβ”€β”€ banana_net_17_12_18.h5
β”‚   β”œβ”€β”€ banana_net_26_02_19.h5
β”‚   └── le_banana_net.h5
β”œβ”€β”€ CNN.py
β”œβ”€β”€ CNN_LeNet.py
β”œβ”€β”€ README.md
β”œβ”€β”€ WebcamApp
β”‚   └── camera_app.py
β”œβ”€β”€ banana_predicter.py
β”œβ”€β”€ datasets
β”‚   β”œβ”€β”€ google_images
β”‚   β”‚   └── ...
β”‚   β”œβ”€β”€ google_synset_fruit365
β”‚   β”‚   └── ...
β”‚   └── synset_fruit365
β”‚       └── ...
β”œβ”€β”€ epoch_fig.png
β”œβ”€β”€ google_images_scraper.js
β”œβ”€β”€ gui
β”‚   β”œβ”€β”€ gui.py
β”‚   └── gui.ui
β”œβ”€β”€ image_downloader.py
β”œβ”€β”€ network_trainer.py
β”œβ”€β”€ predict_tests
β”‚   └── ...
└── urls
    β”œβ”€β”€ google_banana_urls.txt
    └── synset_banana_urls.txt

Datasets

  • Fruit365
  • Synset - ImageNet
  • Google Images

Latest training

EpochFig

About

CNN for recognizing 🍌's in images


Languages

Language:Python 92.0%Language:JavaScript 4.5%Language:Dockerfile 3.5%