armory
armory copied to clipboard
Dataset kwargs for custom dataset pipeline
Would it be possible to add kwargs
to the dataset configuration? Those kwargs
could then be passed along to the dataset_fn
. This would enable us to work around the slow numpy dataset pipeline and integrate our extra features into the pipeline. The check for an ArmoryDataGenerator
in https://github.com/twosixlabs/armory/blob/master/armory/utils/config_loading.py#L42 would need to be changed to a warning for this work.
Would this be just for training or for inference time as well?
Ideally both.