Dataset

Download dataset from https://products-10k.github.io/

Paper at https://arxiv.org/abs/2008.10545

Group

• Christian Bernasconi 816423
• Gabriele Ferrario 817518
• Riccardo Pozzi 807857
• Marco Ripamonti 806785

Bot

Directory structure

The bot expects this directory structure:

.
|__data
|  |__model.h5
|  |__blur_model_4k.h5
|  |__retrieval_base.csv
|  |__train
|     |__1.jpg
|     |  ...
|     |__2629.jpg
|
|__bot
|  |__secrets.py
|
|__indexes
   |__color
   |__hog
   |__neural_network
   |__retrieval_modes.ini

Telegram token

Put the Telegram Bot token inside bot/secrets.py. Look at the sample bot/secrets_sample.py

Requirements

All the requirements needed by the bot are listed inside requirements.txt.

Supposing you prefer to create a virtualenv here are the steps to create a virtualenv and install the requirements:

python -m venv venv

# Linux
# you may need to install some packages; e.g.
# apt-get install ffmpeg libsm6 libxext6
source venv/bin/activate

# Windows
.\venv\Scripts\Activate

pip install -r requirements.txt

Dataset images

Put dataset train images inside ./data/train

Retrieval indexes

Put the indexes inside ./indexes as shown above otherwise create them starting from feature files ./data/color_features.csv ./data/hog_features.csv ./data/nn_features.csv:

once features files are present:

cd scripts
python create_indexes.py

Run

# Get inside the virtualenv if it exists
# Linux
source venv/bin/activate
# Windows
.\venv\Scripts\Activate

# Run the bot
python -m bot