Dataset for CNN regression?

Wondering if there are any datasets available that have image data and a numerical target (not categorical)? I.e. for image-based regression where the target label could be a 1xN numerical tensor.

I’m working on a conditional CVAE to generate images, however my custom dataset is small and I am looking for a large one to validate the architecture on.

Hi @mattbev

Welcome to the PyTorch community!

You can consider object counting datasets, the idea is that object counting can be formulated as a regression problem. Here are some links:

  1. Visual Geometry Group - University of Oxford
  2. Object Counting | Papers With Code
  3. [2008.12470] Counting from Sky: A Large-scale Dataset for Remote Sensing Object Counting and A Benchmark Method
  4. Crowd Counting | Kaggle

Hope this helps!

1 Like