MLDatasets.jl icon indicating copy to clipboard operation
MLDatasets.jl copied to clipboard

Requesting rotated MNIST dataset

Open paulxshen opened this issue 3 years ago • 6 comments

Hello package developers I'd like to request the rotated MNIST dataset. It's a canonical benchmark for testing computer vision algorithms wrt rotated images. We're developing rotation equivariant circular harmonics algorithms that's beating SOTA on some preliminary measures and would appreciate addition of this dataset. Thanks!

paulxshen avatar Sep 02 '22 01:09 paulxshen

Coincidentally I am also working on equivariant neural networks with spherical convolutions and this like a good addition. Unfortunately, I haven't encountered any Julia packages that support equi. networks yet. 😕

Dsantra92 avatar Sep 03 '22 05:09 Dsantra92

@Dsantra92 You can try our EquivariantOperators.jl Tutorials on colab Paper Preprint

paulxshen avatar Sep 12 '22 01:09 paulxshen

This dataset is very easy to implement, one just has to copy the MNIST implementation in the repo and replace the download links with these ones https://github.com/ChaitanyaBaweja/RotNIST/tree/master/data. PR welcome @paulxshen if you have time.

CarloLucibello avatar Sep 12 '22 06:09 CarloLucibello

Thanks will do

paulxshen avatar Sep 17 '22 20:09 paulxshen

It's a little bit more tricky than that. The RotMNIST data points to MNIST data. The repo contains a script to rotate and save them, not difficult to convert the code.

Dsantra92 avatar Oct 02 '22 06:10 Dsantra92

Thanks @Dsantra92 I believe @CarloLucibello dug up the subfolder containing output from the script in the same format as MNIST data

paulxshen avatar Oct 07 '22 15:10 paulxshen