Skip to content

sony/nnabla-examples

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Neural Network Libraries - Examples

This repository contains working examples of Neural Network Libraries. Before running any of the examples in this repository, you must install the Python package for Neural Network Libraries. The Python install guide can be found here.

Before running an example, also run the following command inside the example directory, to install additional dependencies:

cd example_directory
pip install -r requirements.txt

Docker workflow

  • Our Docker workflow offers an easy installation and setup of running environments of our examples.
  • See this page.

nnabla-examples-utils (neu)

neu can now be installed as a python package. It provides a wide range of utility functions. For installation and usage, check utils

Interactive Demos

We have prepared interactive demos, where you can play around without having to worry about the codes and the internal mechanism. You can run it directly on Colab from the links in the table below.

Vision: Generation, Enhancement, Animation

Name Notebook Task Example
SLE-GAN Open In Colab Image Generation
First Order Motion Model Open In Colab Facial Motion Transfer
Zooming Slow-Mo Open In Colab Video Super-Resolution
StyleGAN2 Open In Colab Image Generation
Deep-Exemplar-based-Video-Colorization Open In Colab Video Colorization
TecoGAN Open In Colab Video Super-Resolution
ESR-GAN Open In Colab Super-Resolution
Self-Attention GAN Open In Colab Image Generation
StarGAN Open In Colab Image Translation
DCGAN Open In Colab Image Generation

Vision & Language

Name Notebook Task Example
CLIP Open In Colab Zero-shot image classification

Vision: Recognition

Name Notebook Task Example
CenterNet Open In Colab Object Detection
PSMNet Open In Colab Stereo Depth Estimation
Face Alignment Network Open In Colab Facial Keypoint Detection
YOLO v2 Open In Colab Object Detection
ResNet/ResNeXt/SENet Open In Colab Image Classification

Audio

Name Notebook Task Example
D3Net Open In Colab Music Source Separation
X-UMX Open In Colab Music Source Separation

Machine Learning

Name Notebook Task Example
Out-of-Core training Open In Colab Out-of-Core training
MixUp / CutMix / VH-Mixup Open In Colab Data Augmentation
Virtual Adversarial Training Open In Colab Semi-Supervised Learning
SiameseNet Open In Colab Feature Embedding
Variational Auto-encoder Open In Colab Unsupervised Learning

eXplainable AI

Name Notebook Task Example
Grad-CAM Open In Colab Visualization
SHAP Open In Colab Visualization
Attention Branch Network Open In Colab Visualization

Fairness of Machine Learning

Name Notebook Task Example
Demographic parity
Disparate Impact
Equal opportunity
Equalised odds
Open In Colab [Metrics tutorial]
Dataset/Model Bias Check
Reweighing Open In Colab [Pre-processing tutorial]
Dataset/Model Bias Check and Mitigation by Reweighing
Massage Data Open In Colab [Pre-processing tutorial]
Dataset/Model Bias Check and Mitigation by Massage Data
Preferential Sampling Open In Colab [Pre-processing tutorial]
Dataset/Model Bias Check and Mitigation by Preferential Sampling
GAN Data Debiasing Open In Colab [Pre-processing tutorial]
Dataset/Model Bias Check and Mitigation by GAN
Prejudice Remover Regularizer Open In Colab [In-processing tutorial]
Model Bias Check and Mitigation by Prejudice Removal Technique
Prejudice Remover Regularizer for Images Open In Colab [In-processing tutorial]
Model Bias Check and Mitigation by Prejudice Removal Technique for Images
Adversarial Debiasing Tutorial Open In Colab [In-processing tutorial]
Model Bias Check and Mitigation by Adversarial Debiasing
Adversarial Debiasing for Images Open In Colab [In-processing tutorial]
Model Bias Check and Mitigation by Adversarial Debiasing for Images
Rejection Option based Classification Open In Colab [Post-processing tutorial]
Prediction Bias Check and Mitigation by ROC
Rejection Option based Classification for Images Open In Colab [Post-processing tutorial]
Prediction Bias Check and Mitigation by ROC for Images
Skin color (Masked Images) Open In Colab Facial evaluation for skin color

Model Quantization

Name Notebook Task Example
Post-training quantization Open In Colab Post-training quantization