Skip to content

Images Classification Recipe

CENL-AI-WG edited this page Aug 19, 2020 · 73 revisions

Images Classification Statut

Convolutional Neural Networks (CNN) and Transfer Learning for Heritage Images Classification

This recipe exposes an image classification scenario, aiming to deduce the technique or genre of images (picture, drawing, map...) using a Convolutional Neural Networks trained model (supervised approach with transfer learning).

image classification principle

This recipe may have various library use cases, particularly for cataloguing and information retrieval systems.

This recipe is based on BnF materials

Goals

This recipes includes a basic introduction to neural networks and deep learning (formal neuron model, neural networks, convolutional neural networks, transfer learning) and a hands-on session:

  1. Creation of an images dataset for training
  2. Training of a classification model with commercial APIs or open source IA frameworks
  3. Application of the model to the heritage images to be classified

The IIIF standard API is leveraged to extract images from digital repositories, but raw files may also be used.

Educational resources

Introduction to the core technology used

For the theory, see the BnF github for a 45 mn introduction course (FR and EN versions, direct link).

Implementation notes

A. Images classification using AI SaaS

Prerequisites: IBM Watson Studio account or Google Cloud AutoML account (see the setup document)

1. Use case definition: choice of the source images and the model classes

A 4 classes scenario dataset (picture/drawing/map/noise) can be downloaded here, but it's up to you to build your own use case.

2. Choice of the SaaS platform

IBM Watson Studio and Google Cloud AutoML have been tested and this howto documents the setup of a new user account and the creation of a visual recognition project for both platforms.

The following steps suppose you are using Watson Studio, but the Google AutoML case is very similar.

3. Creation of a Watson Studio Visual Recognition project

Once Watson Studio web app is launched, choose the "Classify Images" custom model to create your new classification project.

4. Download of the training dataset
5. Training of the model
5. Test of the classification model

In the platform

Outside the platform, using code

B. Images classification using AI frameworks

Prerequisites: basic scripting and command line skills (Python scripts are used)

Other resources

https://cloud.google.com/vision/automl/docs/beginners-guide?hl=en ...

Clone this wiki locally