Skip to content

nish996/visual_relationship_detection

Repository files navigation

vrd

Visual relationship detection - Detects relationships between various objects in an image

Preprocessing Data

Generate csv files for training keras-retinat by running csv_gen.py in utils

Train keras-retinanet

Train keras-retinaet by running: python keras-retinanet/keras_retinanet/bin/train.py csv /media/data/nishanth/dataset/train_vrd.csv /media/data/nishanth/dataset/objects.csv

Releases

No releases published

Packages

No packages published