Looking for Pre-trained Model for Image Categorization (Screenshots, Photos, Scans, etc.)

I’m on the lookout for a pre-trained model that’s capable of classifying images into various categories commonly found on a computer. The categories I’m interested in include, but are not limited to:

  • Screenshots
  • Photos
  • Scanned Documents
  • Graphics and Designs
  • Video Thumbnails

My goal is to find a model that understands and categorizes these images based on their nature and content. If a comprehensive model doesn’t exist, I’m also open to models that address subsets of these categories or any advice on how to approach training a model for this purpose.

Here are some specific questions I have:

  • Does anyone know of any pre-trained models that fit expectations?
  • Are there datasets available that could help in training a model for these categories?
  • Any recommendations on model architecture or training strategies for this kind of task?
1 Like