dataset.py role + metadata_compositional-split-natural.t7 structure

ExplainableML / czsl

PyTorch CZSL framework containing GQA, the open-world setting, and the CGE and CompCos methods.

GNU General Public License v3.0

111 stars 27 forks source link

dataset.py role + metadata_compositional-split-natural.t7 structure #19

Closed jiayaohua closed 2 years ago

jiayaohua commented 2 years ago

What is the role of dateset.py?And how is metadata_compositional-split-natural.t7 made?

mancinimassimiliano commented 2 years ago

Hi @jiayaohua,

dataset.py splits the data into training, validation, and test set, processing the input split to get attributes, objects, and compositional information.

metadata_compositional-split-natural.t7 is a list of dictionaries, where each dictionary contains four keys:

image: with the image path
attr: with attribute label
obj: with object label
set: with the set (training, validation, test) to which the image belongs.