armory icon indicating copy to clipboard operation
armory copied to clipboard

Dataset kwargs for custom dataset pipeline

Open dxoigmn opened this issue 4 years ago • 2 comments

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.

dxoigmn avatar May 22 '20 19:05 dxoigmn

Would this be just for training or for inference time as well?

davidslater avatar May 28 '20 17:05 davidslater

Ideally both.

dxoigmn avatar May 28 '20 18:05 dxoigmn