This repo contains code for:
- building a dataset of image prompts with quantities of objects for evaluation
-
the objects are a set of 79/80 objects used in the COCO dataset. https://docs.ultralytics.com/datasets/detect/coco/#dataset-yaml
-
a mini version of the dataset containing ~16k images from a random subset of 10 COCO object classes lives here: https://huggingface.co/datasets/semaj83/ioqm
-
- running various generative models through huggingface diffusers/transformers libraries
- evaluating images generated using the original object detection model developed by facebook for the dataset:
facebook/detr-resnet-50
code to create prompts:
code to generate images:
code to evaluate generated images: