Using MMSEGMENTATION with NPY Files

open-mmlab / mmsegmentation

OpenMMLab Semantic Segmentation Toolbox and Benchmark.

https://mmsegmentation.readthedocs.io/en/main/

Apache License 2.0

8.21k stars 2.6k forks source link

Using MMSEGMENTATION with NPY Files #3474

Open MatCorr opened 10 months ago

MatCorr commented 10 months ago

Has anyone, by any chance, ever implemented code that allows MMSEGMENTATION to train with NPY files rather than images?

I've got really large datasets (which are all larger than a few TBs) that are all in NPY files, and I was hoping I could avoid converting them all to PNG files.

AI-Tianlong commented 9 months ago

I guass u could rewrite LoadImageFromFile and LoadAnnotations

MatCorr commented 9 months ago

Yeah, I'll try to do that. Thanks!

innavoig23 commented 9 months ago

I tried doing that (changing LoadImageFromFile with other functions), but I got errors. Could you tell me if you solve and how?