Skip to content

AntCNN::Improved CNN for Edge Computing Devices and Its Applications on Facial Expression Clasification

Notifications You must be signed in to change notification settings

sureJ0824/AntCNN

 
 

Repository files navigation

Expression-Clasification

AntCNN::Improved CNN for Edge Computing Devices and Its Applications on Facial Expression Clasification

Datasets and AntCNN-G's trained models can be obtained in the following two ways:

1.Google Cloud Disk:https://drive.google.com/drive/folders/1WHebQdZrACVms0_3RDGxTTTSyuzfR48C?usp=sharing 2.Baidu cloud disk:链接:https://pan.baidu.com/s/1mZw9O-cqWhnwH_gu629v2A 提取码:xucr

Please put fer2013_data.h5, RAF_data.h5, CK_data.h5 in the data folder downloaded from the above link into the ./data path.Then, put the RAF_EdgeCNN and FER2013_EdgeCNN folders under the models folder downloaded from the above link into ./models.

That is, the data folder contains the following files:

    ./data:         
            --RAF_data.h5          
            --RAF.py             
            --fer2013_data.h5             
            --fer.py
            --CK_data.h5         
            --CK.py

The models folder contains the following files:

    ./models: 
            --RAF_EdgeCNN folder  
              --PrivateTest_model.t7  
            --FER2013_EdgeCNN
              --PrivateTest_model.t7
            --EdgeNet.py
            --vgg.py
            --resnet.py
            --__init__.py

Experimental environment:

    python 3.6
    pytorch 0.4.0

Running on the Raspberry Pi 3B+:

    pytorch2onn.py: Pytorch cannot be directly converted to an IR file. 
            Therefore, you need to convert the pytorch model to an onxx file using the pytorch2onn.py file.
            Finally, you can convert the onnx file to an IR file on ubuntu.
    
    pi_demo.py: Run on the Raspberry Pi 3B+ using an IR file.

In addition, the modified path of the file is required.

First, we use the dlib library to capture faces. The implementation of this part can be viewed in our other project implementation:https://github.com/tobysunx/face_recognition

About

AntCNN::Improved CNN for Edge Computing Devices and Its Applications on Facial Expression Clasification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 57.2%
  • Jupyter Notebook 42.8%