Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MultiMNIST to MLDatasets? #208

Closed
manuelbb-upb opened this issue Mar 21, 2023 · 1 comment
Closed

Add MultiMNIST to MLDatasets? #208

manuelbb-upb opened this issue Mar 21, 2023 · 1 comment

Comments

@manuelbb-upb
Copy link

To do some experiments along the lines of [1], I reimplemented their data generation routine in Julia.
For MultiMNIST, we want to have two digits in one 28x28 image and 2D labels.
That's why the MNIST type needs to be changed to

struct MultiMNIST <: SupervisedDataset
    metadata::Dict{String, Any}
    split::Symbol
    features::Array{<:Any, 3}
    targets::Matrix{Int}  # `MNIST` has Vector{Int}
end

Here is a constructor to create a MultiMNIST data set from an MNIST object. For that, I use ImageTransformations and Interpolations.

Does it seem like a good idea to try integrating something like that into MLDatasets? It would add those two dependencies...

[1] O. Sener and V. Koltun, “Multi-Task Learning as Multi-Objective Optimization,” arXiv:1810.04650 [cs, stat], Jan. 2019, Accessed: Jan. 24, 2022. [Online]. Available: http://arxiv.org/abs/1810.04650

@manuelbb-upb
Copy link
Author

This will likely better be its own package

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant