Dataset for CNN regression?

vision
1

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.

2

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
  5. http://visal.cs.cityu.edu.hk/static/pubs/conf/cvpr08-peoplecnt.pdf

Hope this helps!

1 Like