ICE-GAN

[our code will be coming soon]

ICE-GAN: Identity-aware and Capsule-Enhanced GAN for Micro-Expression Recognition and Synthesis. (Arxiv version)

Author: Jianhui Yu, Chaoyi Zhang, Yang Song, Weidong Cai

Model Architecture

Installation

The code is tested with Python 3.6, CUDA 10.1, Pytorch 1.4 on Ubuntu 18.04.

Usage

Dataset

• The links for the data we use are provided below:
  1. SMIC dataset
  2. CASME II dataset
  3. SAMM dataset

Classifcation

To train the model from scratch, use the following code:

python main.py

Performance

License