how to save and decode gray-jpeg images?

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

https://ffcv.io

Apache License 2.0

2.81k stars 180 forks source link

how to save and decode gray-jpeg images? #197

Open Nise-2-meet-U opened 2 years ago

Nise-2-meet-U commented 2 years ago

My training data are one channle gray images.

phelps-matthew commented 2 years ago

Yes, it would be great if there were a class similar to ffcv.fields.RGBImageField for grayscale images that allows ones to write them with the same jpeg compression capability.

GuillaumeLeclerc commented 2 years ago

Hi!

There is currently a PR with this functionality. You can try this branch and give feedback in #176 .