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.32k stars 2.63k forks source link

Using MMSEGMENTATION with NPY Files #3474

Open MatCorr opened 11 months ago

MatCorr commented 11 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 10 months ago

I guass u could rewrite LoadImageFromFile and LoadAnnotations

MatCorr commented 10 months ago

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

innavoig23 commented 10 months ago

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