NYC Data Science Academy| Blog
Bootcamps
Lifetime Job Support Available Financing Available
Bootcamps
Data Science with Machine Learning Flagship ๐Ÿ† Data Analytics Bootcamp Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lesson
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories Testimonials Alumni Directory Alumni Exclusive Study Program
Courses
View Bundled Courses
Financing Available
Bootcamp Prep Popular ๐Ÿ”ฅ Data Science Mastery Data Science Launchpad with Python View AI Courses Generative AI for Everyone New ๐ŸŽ‰ Generative AI for Finance New ๐ŸŽ‰ Generative AI for Marketing New ๐ŸŽ‰
Bundle Up
Learn More and Save More
Combination of data science courses.
View Data Science Courses
Beginner
Introductory Python
Intermediate
Data Science Python: Data Analysis and Visualization Popular ๐Ÿ”ฅ Data Science R: Data Analysis and Visualization
Advanced
Data Science Python: Machine Learning Popular ๐Ÿ”ฅ Data Science R: Machine Learning Designing and Implementing Production MLOps New ๐ŸŽ‰ Natural Language Processing for Production (NLP) New ๐ŸŽ‰
Find Inspiration
Get Course Recommendation Must Try ๐Ÿ’Ž An Ultimate Guide to Become a Data Scientist
For Companies
For Companies
Corporate Offerings Hiring Partners Candidate Portfolio Hire Our Graduates
Students Work
Students Work
All Posts Capstone Data Visualization Machine Learning Python Projects R Projects
Tutorials
About
About
About Us Accreditation Contact Us Join Us FAQ Webinars Subscription An Ultimate Guide to
Become a Data Scientist
    Login
NYC Data Science Acedemy
Bootcamps
Courses
Students Work
About
Bootcamps
Bootcamps
Data Science with Machine Learning Flagship
Data Analytics Bootcamp
Artificial Intelligence Bootcamp New Release ๐ŸŽ‰
Free Lessons
Intro to Data Science New Release ๐ŸŽ‰
Find Inspiration
Find Alumni with Similar Background
Job Outlook
Occupational Outlook
Graduate Outcomes Must See ๐Ÿ”ฅ
Alumni
Success Stories
Testimonials
Alumni Directory
Alumni Exclusive Study Program
Courses
Bundles
financing available
View All Bundles
Bootcamp Prep
Data Science Mastery
Data Science Launchpad with Python NEW!
View AI Courses
Generative AI for Everyone
Generative AI for Finance
Generative AI for Marketing
View Data Science Courses
View All Professional Development Courses
Beginner
Introductory Python
Intermediate
Python: Data Analysis and Visualization
R: Data Analysis and Visualization
Advanced
Python: Machine Learning
R: Machine Learning
Designing and Implementing Production MLOps
Natural Language Processing for Production (NLP)
For Companies
Corporate Offerings
Hiring Partners
Candidate Portfolio
Hire Our Graduates
Students Work
All Posts
Capstone
Data Visualization
Machine Learning
Python Projects
R Projects
About
Accreditation
About Us
Contact Us
Join Us
FAQ
Webinars
Subscription
An Ultimate Guide to Become a Data Scientist
Tutorials
Data Analytics
  • Learn Pandas
  • Learn NumPy
  • Learn SciPy
  • Learn Matplotlib
Machine Learning
  • Boosting
  • Random Forest
  • Linear Regression
  • Decision Tree
  • PCA
Interview by Companies
  • JPMC
  • Google
  • Facebook
Artificial Intelligence
  • Learn Generative AI
  • Learn ChatGPT-3.5
  • Learn ChatGPT-4
  • Learn Google Bard
Coding
  • Learn Python
  • Learn SQL
  • Learn MySQL
  • Learn NoSQL
  • Learn PySpark
  • Learn PyTorch
Interview Questions
  • Python Hard
  • R Easy
  • R Hard
  • SQL Easy
  • SQL Hard
  • Python Easy
Data Science Blog > Capstone > Image Classification Improved Using CNN, MotionFlow

Image Classification Improved Using CNN, MotionFlow

Jonathan Presley and Youngmin Paul Cho
Posted on Jun 19, 2019

Project GitHub | LinkedIn:   Niki   Moritz   Hao-Wei   Matthew   Oren

The skills we demoed here can be learned through taking Data Science with Machine Learning bootcamp with NYC Data Science Academy.

Introduction

Billions of dollars are wasted on digital ads that are ultimately unseen, sipped, ignored, or closed.  That is why MotionFlow, the partnering company for our capstone project, wants to leverage computer vision and the Internet of Things (the connectivity of electronic, everyday devices) to provide targeted ads in offline and more attentive spaces. In order for these ads to be personalized, the computer needs to be able to identify characteristics of the individual in real time.  Computer vision is advancing rapidly, and now image classification works better than ever before.

For our capstone project, our task was to provide a model which can predict/classify age range, race, and gender, given an image of a personโ€™s face.  As many computer vision researchers do, we approached with convolutional neural network method (CNN) with Python using TensorFlowโ€™s Keras package to produce a model which predicts those characteristics with 50, 76, and 92 percent accuracy, respectively, and output a JSON file of the results.  

In this blog post, we will explain our approach by studying existing CNN architectures, illustrate our model and results, and give some practical advice for deep learning.

 

If you need a bit of review on deep learning concepts and CNNs, weโ€™ve compiled some resources that might be helpful.

View our code files on our github accounts (Paul and Jonathan)

Dataset of Images:

We have used dataset provided from our partnering company, MotionFlow, as well as face dataset found from UTKFace. Our dataset comprised more than 23,000 images with labeled age, gender, and ethnicity.  Each file name included the true value of age, gender, and ethnicity of the person in the image.  

Labels:

The labels of each face image is embedded in the file name, formatted:

[age_gender_race_date&time].jpg

  • age: an integer from 0-116
  • gender: either 0 (male) or 1 (female)
  • race: an integer from 0 to 4, denoting Caucasian, Black, Asian, Indian, and Hispanic
  • date&time: format of yyyymmddHHMMSSFFF

 

Exploratory Analysis

In order to get a better sense of the balance and spread of our set of 23,000 images, we needed to parse the filenames and store them into a dataframe for analysis.  

Our dataframe included columns for the file pathway, age, race, and gender. When grouped by race and gender, we found that there are far more images that are labeled as โ€œwhite maleโ€ and โ€œwhite femaleโ€ than the other groups.   In addition, the differences in count between male and female were relatively similar.

Count of images grouped by race and age.

When looking at the histogram for age overall, itโ€™s clear that there are many images of people in their mid-twenties and early thirties.  We also made density plots of age after they were grouped by gender and race, which revealed that infants were also proportionally over-represented in the data set.

 

 

Knowing that infants arenโ€™t the target audience in practical terms for MotionFlow (ads will most likely not be intended to be read by the infants themselves) and it might be difficult to distinguish features which identify gender and possibly race, we did not include pictures of ages under seven.  Next, we made a function that took the continuous values of age and grouped them into age groups, seen below. When coded, age group became an ordinal variable.

Generally, itโ€™s a good idea to balance the dataset across all categories when training a model so that itโ€™s not trained expecting a certain proportion of one category.  However, balancing the data across all gender, age, and race categories by undersampling would make the data set too small because there are some combinations of categories with only 8 or 9 pictures.  

Thereโ€™s a method called data augmentation, which distorts, rotates, or flips the images to create more samples, but this technique alone on some granular subcategories (like 75+, other, male) would not solve the data size issue.  Therefore, we just undersampled the white category by 50 percent at random and saw how our model performed with a somewhat unbalanced data set. Below is the bar plot of counts grouped by race and gender after the white category reduction.

Count of images subcategorized by race and gender before and after undersampling.

Before we use the JPG pixel values as input values for the CNN model, simple normalization is performed. This is done because we want to ensure equivalent effect on every dimension when doing backpropagation and it also converges much faster from the gradient error.

Model

Before creating our own model, we identified the key benefits and strengths of each structures of architectures which have done well in prior ImageNet competitions (image classification competition) including LeNet-5, AlexNet, VGGNet, and ResNet.

Looking at the architecture of these models, we noticed that we noticed that as accuracy improved these models increased the number of layers and complexity.  For our model, we combined the uniform and repetitive convolution layers of VGGNet model with the use of dropouts from AlexNet. As seen below, our model uses 6 convolution cycles, each consisting of a convolution, batch normalization, and max pooling layer, and two dense layers with dropout.

Our base CNN model architecture.

As our data flows through the network of Conv2D layer, the weights and parameters may get too big or too small again. Therefore we set a batch normalization layer in between to normalize our data and converge much faster when training layer.

We used max pooling following the batch normalization to reduce the dimensions of the input volume for the next convolutional layer. Its function is to reduce the amount of parameters and reduce the computational burden of the network.

We tried a few variations of our model using ReLU as our activation function, one using LeakyReLU, and one replacing our convolution layers with a pre-trained AlexNet model with set weights. The results are shown in the following sections.  

RESULTS

 

ReLU with (Conv -> BatchNorm -> MaxPool) * range(3,6)

The first model we tried used the same architecture as shown above, using ReLU as the activation function and the dropout rate was 50 percent. The details of each layer is shown in the image below.  There were over one million trainable parameters.

As seen from the output below, the accuracy the race, gender, and age for the validation set was around 70, 90, and 47 percent respectively after 14 epochs.

This was not too bad, but we wanted to see how this could be improved.

LeakyReLU with (Conv -> BatchNorm -> MaxPool) * range(3,6)

Our results using LeakyReLU were slightly better overall, but the accuracy did not continually improve after a handful of epochs like we saw in the previous model.  The values of the validation accuracy remained around 76, 92, and 50 percent for race, gender, and age respectively.

Next, we wanted to see how transfer learning could be used to our advantage, using pre-trained weights from an existing model to jumpstart the training process and hopefully improve the accuracy of our model.

Transfer Learning Using ResNet

As seen above in the figure comparing the performance of famous models, ResNet architecture has significantly improved performance. The key to being able to train such a very deep network is to use skip connections. the network will be forced to model f(x) = h(x) โ€“ x rather than h(x).  This is called residual learning.

We were able to load a pre-trained ResNet model into our own modelโ€™s architecture and train our model, only adjusting the weights of our dense layers at the end.

Surprisingly, the ResNet transfer learning model performed quite poorly, only achieving 27, 52, and 11 percent accuracy for the validation set for race, gender, and age respectively.  

Below is the 32 example outputs as in image format. Prediction is written on top and actual is presented on the bottom.

 

Conclusion and Next Steps

Our best performing model, using our own CNN architecture and LeakyRelu, achieved 75, 92, and 48 percent accuracy for predicting race, gender, and age respectively.  Here are some next steps that would possibly help improve the modelโ€™s performance:

  • Try other combinations of data transformations, percentage of dropout, activation functions, and see which combinations work well.
  • Train three separate models for each type of output.
  • Use grayscale images instead of color images.
  • Balance the training data set further.

 

Looking back on our approach, it might have been more appropriate to train three separate models for each of the desired targets.  This would have solved some of the data limitation problems when balancing the data set because separating the images by label would have been easier if age, race, and gender were treated independently.  Also, it was suggested to us that using grayscale images and further balancing of the dataset would have resulted in a better performance.

Practical Tips

Setting Up Your Environment

TensorFlow updates its versions frequently and often makes code from earlier versions incompatible with the new version.  To save yourself headache, set up a different local environment using Anaconda Navigator. For example, if you want to follow along with a YouTube tutorial and they are using version 2.0 but you already installed version 1.13.1, just make a new environment and install TensorFlow 2.0.  Here are instructions for how you make a new environment.  That way you can switch back and forth between projects and environments without having to use or set up a Docker image.

Training Speed

Training CNNs on image data is notoriously slow.  Using Jonathanโ€™s MacBook Pro, each epoch took over an hour.  Luckily, Paul had a desktop which used GPUs to train our model in which each epoch was about five minutes.  Alternatively, we could have used the cloud computing services of Paperspace or AWS.  The price is relatively low, but you still want to use your time wisely and be well-prepared in knowing what versions of the model you want to train.

We hope you have enjoyed reading about our project and learned a bit more about image classification using CNNs.  Feel free to leave us comments below or check out our other blog posts by clicking the author links.

About Authors

Jonathan Presley

Jonathan Presley, MPH is a data scientist fellow dedicated to efficiency, community wellness, and service. With nine years of combined experience in scientific research, biotechnology, education, and healthcare, he has developed a variety of skills in research, policy,...
View all posts by Jonathan Presley >

Youngmin Paul Cho

Paul earned a B.S. in Mathematical Statistics from the University of Texas at Austin. His recent work has been related to Process Innovation, Web-databse Development, and SCM. He likes the challenge of drawing on creative solutions to problems...
View all posts by Youngmin Paul Cho >

Leave a Comment

Cancel reply

You must be logged in to post a comment.

No comments found.

View Posts by Categories

All Posts 2399 posts
AI 7 posts
AI Agent 2 posts
AI-based hotel recommendation 1 posts
AIForGood 1 posts
Alumni 60 posts
Animated Maps 1 posts
APIs 41 posts
Artificial Intelligence 2 posts
Artificial Intelligence 2 posts
AWS 13 posts
Banking 1 posts
Big Data 50 posts
Branch Analysis 1 posts
Capstone 206 posts
Career Education 7 posts
CLIP 1 posts
Community 72 posts
Congestion Zone 1 posts
Content Recommendation 1 posts
Cosine SImilarity 1 posts
Data Analysis 5 posts
Data Engineering 1 posts
Data Engineering 3 posts
Data Science 7 posts
Data Science News and Sharing 73 posts
Data Visualization 324 posts
Events 5 posts
Featured 37 posts
Function calling 1 posts
FutureTech 1 posts
Generative AI 5 posts
Hadoop 13 posts
Image Classification 1 posts
Innovation 2 posts
Kmeans Cluster 1 posts
LLM 6 posts
Machine Learning 364 posts
Marketing 1 posts
Meetup 144 posts
MLOPs 1 posts
Model Deployment 1 posts
Nagamas69 1 posts
NLP 1 posts
OpenAI 5 posts
OpenNYC Data 1 posts
pySpark 1 posts
Python 16 posts
Python 458 posts
Python data analysis 4 posts
Python Shiny 2 posts
R 404 posts
R Data Analysis 1 posts
R Shiny 560 posts
R Visualization 445 posts
RAG 1 posts
RoBERTa 1 posts
semantic rearch 2 posts
Spark 17 posts
SQL 1 posts
Streamlit 2 posts
Student Works 1687 posts
Tableau 12 posts
TensorFlow 3 posts
Traffic 1 posts
User Preference Modeling 1 posts
Vector database 2 posts
Web Scraping 483 posts
wukong138 1 posts

Our Recent Popular Posts

AI 4 AI: ChatGPT Unifies My Blog Posts
by Vinod Chugani
Dec 18, 2022
Meet Your Machine Learning Mentors: Kyle Gallatin
by Vivian Zhang
Nov 4, 2020
NICU Admissions and CCHD: Predicting Based on Data Analysis
by Paul Lee, Aron Berke, Bee Kim, Bettina Meier and Ira Villar
Jan 7, 2020

View Posts by Tags

#python #trainwithnycdsa 2019 2020 Revenue 3-points agriculture air quality airbnb airline alcohol Alex Baransky algorithm alumni Alumni Interview Alumni Reviews Alumni Spotlight alumni story Alumnus ames dataset ames housing dataset apartment rent API Application artist aws bank loans beautiful soup Best Bootcamp Best Data Science 2019 Best Data Science Bootcamp Best Data Science Bootcamp 2020 Best Ranked Big Data Book Launch Book-Signing bootcamp Bootcamp Alumni Bootcamp Prep boston safety Bundles cake recipe California Cancer Research capstone car price Career Career Day ChatGPT citibike classic cars classpass clustering Coding Course Demo Course Report covid 19 credit credit card crime frequency crops D3.js data data analysis Data Analyst data analytics data for tripadvisor reviews data science Data Science Academy Data Science Bootcamp Data science jobs Data Science Reviews Data Scientist Data Scientist Jobs data visualization database Deep Learning Demo Day Discount disney dplyr drug data e-commerce economy employee employee burnout employer networking environment feature engineering Finance Financial Data Science fitness studio Flask flight delay football gbm Get Hired ggplot2 googleVis H20 Hadoop hallmark holiday movie happiness healthcare frauds higgs boson Hiring hiring partner events Hiring Partners hotels housing housing data housing predictions housing price hy-vee Income industry Industry Experts Injuries Instructor Blog Instructor Interview insurance italki Job Job Placement Jobs Jon Krohn JP Morgan Chase Kaggle Kickstarter las vegas airport lasso regression Lead Data Scienctist Lead Data Scientist leaflet league linear regression Logistic Regression machine learning Maps market matplotlib Medical Research Meet the team meetup methal health miami beach movie music Napoli NBA netflix Networking neural network Neural networks New Courses NHL nlp NYC NYC Data Science nyc data science academy NYC Open Data nyc property NYCDSA NYCDSA Alumni Online Online Bootcamp Online Training Open Data painter pandas Part-time performance phoenix pollutants Portfolio Development precision measurement prediction Prework Programming public safety PwC python Python Data Analysis python machine learning python scrapy python web scraping python webscraping Python Workshop R R Data Analysis R language R Programming R Shiny r studio R Visualization R Workshop R-bloggers random forest Ranking recommendation recommendation system regression Remote remote data science bootcamp Scrapy scrapy visualization seaborn seafood type Selenium sentiment analysis sentiment classification Shiny Shiny Dashboard Spark Special Special Summer Sports statistics streaming Student Interview Student Showcase SVM Switchup Tableau teachers team team performance TensorFlow Testimonial tf-idf Top Data Science Bootcamp Top manufacturing companies Transfers tweets twitter videos visualization wallstreet wallstreetbets web scraping Weekend Course What to expect whiskey whiskeyadvocate wildfire word cloud word2vec XGBoost yelp youtube trending ZORI

NYC Data Science Academy

NYC Data Science Academy teaches data science, trains companies and their employees to better profit from data, excels at big data project consulting, and connects trained Data Scientists to our industry.

NYC Data Science Academy is licensed by New York State Education Department.

Get detailed curriculum information about our
amazing bootcamp!

Please enter a valid email address
Sign up completed. Thank you!

Offerings

  • HOME
  • DATA SCIENCE BOOTCAMP
  • ONLINE DATA SCIENCE BOOTCAMP
  • Professional Development Courses
  • CORPORATE OFFERINGS
  • HIRING PARTNERS
  • About

  • About Us
  • Alumni
  • Blog
  • FAQ
  • Contact Us
  • Refund Policy
  • Join Us
  • SOCIAL MEDIA

    ยฉ 2025 NYC Data Science Academy
    All rights reserved. | Site Map
    Privacy Policy | Terms of Service
    Bootcamp Application