• Stars
    star
    156
  • Rank 239,589 (Top 5 %)
  • Language
    Jupyter Notebook
  • Created over 4 years ago
  • Updated over 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Repo for 42 days project to replicate/improve Airbnb's amenity (object) detection pipeline.

Replicating Airbnb's Amenity Detection with Detectron2 👁 🛏

This repository contains all the code from a 42-day project to replicate Airbnb's amenity detection using Detectron2.

example of how amenities might be detected in an image

Sample image taken from: https://www.airbnb.com/rooms/2151100

What's amenity detection?

Object detection but for common and useful household items someone looking for an Airbnb rental might want to know about. For example, does this home have a fireplace?

Original inspiration was drawn from the article Amenity Detection and Beyond — New Frontiers of Computer Vision at Airbnb.

Try it for yourself

To see the major pieces of the puzzle (data downloading, preprocessing, modelling), check out the example Colab Notebook.

You can also see a live and deployed app which uses the model (note: this may have been taken down by the time you check it due to costs, an extension would be finding a way to deply the model for little to no costs).

If the deployed application doesn't work, you can watch this video to get an idea:

deploying the airbnb amenity detection machine learning app on youtube

What's in this repo?

  • Notebooks with 00-10 are all the steps I took to train the full model, largely unchanged from when I originally wrote them.
  • preprocessing.py contains the preprocessing functions for turning Open Images images & labels into Detectron2 style.
  • downloadOI.py is a slightly modified downloader script from LearnOpenCV which downloads only certain classes of images from Open Images, example:
# Download only images from the Kitchen & dining room table class from Open Images validation set
!python3 downloadOI.py --dataset "validation" --classes "Kitchen & dining room table"
  • app contains a Python script with a Streamlit app built around the model, if you can see the live version, Streamlit is what I used to build it.
  • custom_images contains a series of different images related to the project, including various rooms around my house to test the model.

See how it was done

Questions/feedback/advice is welcome: daniel at mrdbourke dot com.

More Repositories

1

machine-learning-roadmap

A roadmap connecting many of the most important concepts in machine learning, how to learn them and what tools to use to perform them.
7,121
star
2

pytorch-deep-learning

Materials for the Learn PyTorch for Deep Learning: Zero to Mastery course.
Jupyter Notebook
6,701
star
3

tensorflow-deep-learning

All course materials for the Zero to Mastery Deep Learning with TensorFlow course.
Jupyter Notebook
4,709
star
4

zero-to-mastery-ml

All course materials for the Zero to Mastery Machine Learning and Data Science course.
Jupyter Notebook
2,449
star
5

cs329s-ml-deployment-tutorial

Code and files to go along with CS329s machine learning model deployment tutorial.
Jupyter Notebook
573
star
6

m1-machine-learning-test

Code for testing various M1 Chip benchmarks with TensorFlow.
Jupyter Notebook
482
star
7

pytorch-apple-silicon

Setup PyTorch on Mac/Apple Silicon plus a few benchmarks.
Jupyter Notebook
356
star
8

your-first-kaggle-submission

How to perform an exploratory data analysis on the Kaggle Titanic dataset and make a submission to the leaderboard.
Jupyter Notebook
216
star
9

nutrify

Take a photo of food and learn about it.
Jupyter Notebook
156
star
10

mac-ml-speed-test

A few quick scripts focused on testing TensorFlow/PyTorch/Llama 2 on macOS.
Jupyter Notebook
86
star
11

food-not-food

Machine Learning powered app to decide whether a photo is food or not.
Jupyter Notebook
54
star
12

python-list-comprehensions-tutorial

This is the code to go along with the Python list comprehensions video by Daniel Bourke on YouTube.
Jupyter Notebook
43
star
13

mrdbourke

34
star
14

learn-transformers

Work in progress. Simple repository to learn Transformers (and transformers).
Jupyter Notebook
28
star
15

old_blog

My dev's blog
HTML
21
star
16

rag-resources

A collection of curated RAG (Retrieval Augmented Generation) resources.
21
star
17

charlie-walks

Website code for Charlie Walks: A Novel by Daniel Bourke
HTML
12
star
18

Udacity_DLND_Projects

All of my projects from the Udacity Deep Learning Foundations Nanodegree.
Jupyter Notebook
12
star
19

coursera_bioinformatics_and_genetic_algorithm_experiment

Code relating to the Coursera Bioinformatics Specialization as well as my own genetic algorithm experiment.
Jupyter Notebook
11
star
20

AIND-Machine-Translation

The code and other files related to the Udacity Artificial Intelligence Nanodegree Machine Translation project.
Jupyter Notebook
10
star
21

twitch-ml-deploy

Deploying a ML model in the browser using TensorFlow JS
JavaScript
8
star
22

Sentiment-Analysis-with-Keras

Materials and code relating to Learning Intelligence 25.
Jupyter Notebook
8
star
23

AIND-VUI-Capstone

Code and other files related to the Udacity Artificial Intelligence Nanodegree Deep Neural Network Speech Recognizer.
HTML
7
star
24

udacity-AIND

All of my files and projects associated with Term 1 of the Udacity Artificial Intelligence Nanodegree.
HTML
5
star
25

LearnPythonTheHardWay

All of the exercises from the Learn Python the Hard Way book from Zed Shaw.
Python
5
star
26

pytorch-resnet-twitch

PyTorch implementation(s) of various ResNet models from Twitch streams.
Jupyter Notebook
5
star
27

udacityDLfoundationsP2

Second project of the Udacity Deep Learning Foundations Nanodegree
HTML
5
star
28

nutrify-ai-grant-application

Nutrify application to aigrant.org
HTML
4
star
29

AIND-NLP

Materials relating to the Udacity AIND NLP classes.
Jupyter Notebook
3
star
30

modal-test

testing modal.com
Python
2
star
31

Treehouse

Projects from my Treehouse Coursework
2
star
32

hello-world

Github practice
2
star
33

food-data-central-database-creation-with-supabase

JavaScript
2
star
34

html-search-bar

Simple HTML search bar to search through a JavaScript array of JSON.
JavaScript
2
star
35

iOS-Retro-Calculator

Retro/Space
1
star
36

test-data-size-repo

1
star
37

ios-course-super-cool-app

Basic-iOS-app
1
star
38

blog-starter-test-1

🚀⚡️ Blazing fast blog built with Gatsby and Cosmic JS 🔥
JavaScript
1
star