search

JuliaML / MLDatasets.jl

Utility package for accessing common Machine Learning datasets in Julia
https://juliaml.github.io/MLDatasets.jl/stable
MIT License
224 stars 44 forks source link

Requesting rotated MNIST dataset #171

Open paulxshen opened 1 year ago

paulxshen commented 1 year ago

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!

Dsantra92 commented 1 year ago

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. 😕

paulxshen commented 1 year ago

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

CarloLucibello commented 1 year ago

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.

paulxshen commented 1 year ago

Thanks will do

Dsantra92 commented 1 year ago

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.

paulxshen commented 1 year ago

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