This repository contains the second part for generating features of the challenge "KDD BR Competition 2022" of the team Artificial Psycho Killer.

numpy==1.21.6
pandas==1.3.5
keras==2.6.0
opencv-contrib-python==4.5.4.60
opencv-python==4.5.4.60
opencv-python-headless==4.5.4.60

Can be found in requirements.txt file.

The input data needed to generate the features are the same distributed by the competition:

• public.csv
• train images
• test images

Running notebooks kddbr22 Img Shift.ipynb and kddbr20-flann-based-matcher.ipynb will output the following files:

• mario_train_vec_v1.csv
• mario_test_vec_v1.csv
• vecs_v3_fe.csv

These three files are inputs for the next steps of the team's solution.

Before running the notebooks you may want to change the input/output paths to match your directories structure.