clementsan / cnn-image-classification

AI-based image classification tasks (via CNN networks)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

AI-based image classification tasks on Scanning Elecron Microscope (SEM) data

Analysis tasks

Preprocessing scripts

  • Generating small tiles from large images

Image classification

  • Image classification via pytorch

  • Image classification via fastai

    • Direct classification on small tiles
    • Classification with Majority Voting
    • Classification with 5-Fold Cross Validation (CV)

Notes

Neural networks

Use of various ResNet models

Deep learning libraries

Use of pytorch and fastai libraries

About

AI-based image classification tasks (via CNN networks)


Languages

Language:Jupyter Notebook 83.2%Language:Python 16.3%Language:Shell 0.6%