h3 / Popbot

Let AI transform your images 🤖

Popbot

Let AI transform your images

Installation: (run both client & server simultaneously)

client:

cd client

# Install dependencies
npm install

# Start dev server
npm run serve

server:

IMPORTANT: Download the model file on this link and put it in the server directory before doing the following.

cd server
pipenv shell

# Install dependencies (using Pipenv)
sudo pipenv install --skip-lock

# Start server with a category to detect
python main.py <category>

# Example:
python main.py person

Categories:

You can start the server with one of these categories as arguments to detect them in your image.

1. Vehicles:

• car
• bicycle
• motorcycle
• airplane
• bus
• train
• truck
• boat

2. Animals:

• bird
• cat
• dog
• horse
• sheep
• cow
• elephant
• bear
• zebra
• giraffe

3. Food:

• banana
• apple
• orange
• carrot
• pizza
• cake
• cup
• bottle

4. Other:

• person
• umbrella
• skateboard
• laptop
• tv
• book

NOTES:

• For some weird reason, you have to restart the server in order to transform another image. If you don't, you'll get the first image as output. I didn't have time to figure out why that happens.
• You can't combine two categories.
• By default, only the largest object will be detected.

Demo: