Skip to content

fulviomascara/pytorch-cv

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

PyTorch for Computer Vision
Example 1 - Convolutional Bebezal Network

After this notebook, you'll be able to:

  • Use a folder with labels as subfolders as a dataset
  • Split train dataset into train, test and val datasets
  • Apply transformations in images for data augmentation
  • Create a Convolutional Neural Network
  • Train and validate your Convolutional Neural Network
  • Analyze metrics obtained in training task
  • Run inference in test dataset

TO DO

  • Detect faces in a image, using another CNN
  • Train and validate, using only face detected

The architecture of the neural network created in this notebook is based on paper "Facial Expression Recognition Research Based on Deep Learning" by Yongpei Zhu, Hongwei Fan, Kehong Yuan

Link: https://arxiv.org/pdf/1904.09737v1.pdf

CN

About

Pytorch for Computer Vision

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published