Skip to content

Added script for extracting pedestrian images and negative images from KITTI dataset

Valentina Kustikova requested to merge github/fork/VladVin/image-preproc into master

Created by: VladVin

Добавлены 2 скрипта: extract_kitti_pedestrians.py для извлечения позитивов согласно аннотации и extract_kitti_negatives.py для извлечения негативов из изображений датасета KITTI. Изображения сохраняются в .png файлы в заданную директорию.

Скрипт extract_kitti_pedestrians.py выполняет подгонку размеров выходного изображения в соответсвии с заданными параметрами. Подгонка осуществляется следующим образом: из аннотации берутся параметры прямоугольника, из исходного изображения вырезается этот прямоугольник с дополнением недостающих до пропорции w / h пикселей, после этого выполняется масштабирование до заданного размера. Использование скрипта:

python extract_kitti_pedestrians.py first second third forth
first - input images dir
second - annotation files dir
third - output images dir (where you want to store pedestrian images)
forth - desired size in format wxh. For example: 36x108'

Скрипт extract_kitti_negatives.py вырезает негативы из исходных изображений. Использование скрипта:

python extract_kitti_negatives.py first second third forth
first - input images dir
second - output images dir (where you want to store negative images)
third - desired size in format wxh. For example: 36x108
forth - count of negatives to extract

Merge request reports