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 > Data Object Localization using Bounding Boxes Deep Learning

Data Object Localization using Bounding Boxes Deep Learning

Sabbir Mohammed
Posted on May 18, 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.

An Image Processing Project based on Neural Network Deep Learning 


Sponsored by Koios Medical, this project aims to build a deep learning neural network model that can localize potential tumors and/or lesions on ultrasound scans of the thyroid. This model is trained on a data set of patient thyroid scans that have already been marked by radiologists, available here (based on this research paper). The model is built on the Keras framework (using tensor flow on the backend) and utilizes the Xception pre-trained weights to initialize its training.

The code for this project can be found here.

Project Background


This capstone project for the NYC Data Science Academy was sponsored by Koios Medical. The project was aimed at developing a reliable model that can essentially automate the particular function of the radiologist, whereby they mark/ identify potential lesions or tumors for further biopsy or investigation.  The envisioned benefit of using a machine learning method is that it can possibly identify areas of interest that might be missed by a human radiologist due to image quality, error or any other reason, while also increasing efficiency in reporting test results to physicians. The model itself, however, was designed to train for this task using marks already made by radiologists from the publicly available data set linked above.

 

Project Data and Image Augmentation


The data in this project consists of black and white images ( grayscale with three "color-channels") of thyroid ultrasound scans, that were sized either (315 x 560) or (360 x 560) pixels. There were also .csv files that assigned image IDs to each scan and the corresponding directory path, as well as bounding box coordinates (upper left-hand side x and y values and the weight and height) alongside pixel values/ positions for segmentation purposes (alternate method of object localization).

There are 466 raw images from the dataset that were eventually used in this project, which is a rather scare quantity. Therefore, the first step was to augment these images to generate new data for modeling. The "imgaug" open source Python library was used to generate them. The following code snippet outlines the transformation parameters that were applied to the images:

Set of Image augmentation parameters. Relevant Imgaug documentation

 

To elaborate, 50% of the image set was randomly chosen to be horizontally flipped/ mirrored into new images. Additionally, the image contrast was varied randomly between 75% and 150% of the original contrast,, as well as brightness (between 80% and 120%) and finally, the rotation transformation between (-15 and 15 degrees).

Some of the scans in the dataset were "double" images, an example of which can be found below:

Example of a "double" image scan

 

Some of the limitations of the structure of this project and the model used meant that only one bounding box can be predicted per image. It is most definitely a severe limitation of this project which will be discussed later. However, at this point in the augmentation process, to maintain simplicity in the dataset, these "double" images were removed. There were 172 of them, thus reducing the 466 image set to 294.

The imgaug library allows the user to determine the number of new images to be augmented based on a multiplied factor of the original set of images. The final, augmented data set therefore included 3,234 images (294 original and 294 x 10 = 2,940 augmented). This was split into 60-20-20 training, validation and testing sets (3,175, 1,058 and 1,059 images respectively)

A sample augmentation is displayed below (the original and transformed image):

Pre-Marked Bounding Box on Ultrasound Scan

Translated Bounding Box on Augmented image (with transformations applied to the same prior image)

 

Project Model Architecture


The neural network for this project was compiled, trained and tested using the Keras framework (using TensorFlow in the backend), as mentioned above.  In fact, the Xception model was used to build this prediction model, including loading the pre-trained weights from ImageNet. Details can be found here. The following image is an outline of the layers that comprise of Xception. Clicking the image will lead to a concise explanation of this model.

 

Xception Neural Network layers

 

While these models are typically used for object detection using a classification-problem approach, since our goal was simplified to consider only object localization prior to classification, as an exercise in learning and experimentation, the Xception model was modified to take on a regression-problem approach. In this case, the network seeks to "regress" the bounding box coordinates onto the images and predict them based on fitting to the pre-defined target variable (the set of coordinates in the dataset).

For the purposes of this project, once Xception was loaded on as the base layers (with an input image size of 299 x 299 x 3):

  • A Flatten layer was added to begin extracting the bounding box coordinates after the images were analyzed and "seen" by the model.
  • This was followed by a 50% Dropout layer to ease the computational burden from this model.
  • Finally, two Dense layers were added to reduce the output dimensions down to 64, then finally 4 to correspond to the four points of the bounding box coordinates (technically two points: upper-left and lower-right corners). The Dense layers utilized a linear activation function.

The following is a code output of the model summary:

Model layers implemented (for this project specifically) after pre-trained Xception architecture was already loaded

 

Since Keras was being used, it allowed the creation of training, validation and test data "generators" that Keras then uses for fitting, optimizing and predicting. The Keras image-data-generators were built based on the flow_from_dataframe method - instead of using a set directory for the predictor or "feature" images, a "directory" variable (column) in the data frame corresponds to each image and the rows indicate the bounding box coordinates.

 

Project Model Performance


The model isn't particularly efficient and computationally very heavy, both due to the structure and code, as well as the nature of the problem itself. Processing over 3,000 images means a regular, personal laptop is not sufficient. As such, the project required the use of GPUs which were available for public use through the Google Colab service. 

The neural network was fitted on the training and validation image sets over 25 epochs, the results of which are available below. This model was evaluated on two metrics. The first is a mean-squared error (MSE) loss, and the second is the IoU (Intersection over Union). A thorough explanation of IoU can be found here.  The IoU metric had to be manually created as a TensorFlow and Python function since it is not a built-in metric for Keras. The two sources that were combined and used to this purpose can be found here and here.

On the first graph, we can see that the training and validation IoU scores are closely matched and they converge to a stabilized score of roughly 80% before the 10th epoch. In the second graph, we see the MSE  loss drop to values close to zero from the 2nd epoch onwards. These results seem suspicious.

The first issue is that the validation scores are actually better than the training scores. This is a strong indicator of overfitting, however, some brief research has shown that this is a known consequence of the use of a Dropout layer (since a dropout layer removes observations during training, but leaves data in place during validation).

The next issue is the tremendously low MSE loss, which might still indicate overfitting and a possible case of data leakage. However, the results of the test set predictions might be more indicative.

Training and Test set performances over 25 epochs. IoU scores (above) and MSE (mean-squared error) loss (below)

 

The average IoU score from the predictions on the test set turned out to be 42.3% with the highest score of 78.4%. The result is definitely far from optimal, but a closer look at the outcomes show that the predictions are definitely around the correct location on the image, however, the model falls short in being able to correctly predict the lesion size, hence the existing score. 

Below are some sample outputs from the model prediction. The first image is the best IoU achieved. The rest are randomly chosen:

 

Project Conclusions and Future Work


There is definitely a lot of improvements to be made on this project and the model can certainly be tested with other parameters and changes. For example, the severe limitation of being able to predict only one set of bounding boxes per image, the resource-intensive structure of the project, the scarcity of data/ images, etc. 

One of the changes that were considered was converting all the coordinate values from absolute values (pixels) to scaled values based on the image dimensions so that they would be bound between 0 and 1 in order to cater to TensorFlow's preference for small numbers. Additionally, a sigmoid activation function could be used on the dense layers instead of a linear one - as a test to see if it lead to better, more stabilized results.

However, as a first-time foray into deep learning and neural networks in general, this was a tremendous learning experience with tangible results. The goal is to continue learning and building on this progress and fine-tuning the techniques behind the project to achieve better and useful results. 

 


Thank you so much for viewing my project!

  -  All Comments and Suggestions are Welcome!  - 

About Author

Sabbir Mohammed

Sabbir is an aspiring data scientist with a recent certification from the NYC Data Science Academy. He obtained his BS in Mechanical Engineering from Rensselaer Polytechnic Institute and has since spent several years in logistics and procurement for...
View all posts by Sabbir Mohammed >

Related Articles

Community
Data Analysis: Real Estate and Machine Learning
AWS
A.I. Development for Two Sigma Halite II Challenge
Capstone
Predicting Profit Warnings: NLP applied to Conference Call Transcripts Analysis
Student Works
Tribots read and write
APIs
Building a Painting Style Classifier with Tensorflow

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