See https://github.com/autognc/starfish.
This repo contains scripts used to generate labeled, synthetic imagery of Cygnus and Gateway by leveraging Starfish.
- gen_cygnus_dataset.py: For an example '.yaml' see sample_config.yml. This script is used to generate multiple imagesets one after another. Each imageset can have an array of different augmentations. Great for creating datasets with multiple imagesets of various sizes with glare, blur, occlusion, or background randomization(or any combination of these augmentations). Images are labeled with bboxes and keypoints. NOTE: Background randomization technique depends on the .blend file used(see line 231 of script).
- Interpolated_cygnus_GB.py & Interpolated_dynamic.py: This script is used for creating interpolated image sequences with glare and blur of Cygnus and Gateway respectively.
- cygnus_RT.py: This script is used to render cygnus images with randomized textures.
- cygnus_keypointsGB.py: This script is used to render augmented cygnus images labeled with bboxes and keypoints. This script generates a single imageset, and has the same augmentation options as gen_cygnus_dataset.py
- cygnus_occlusion_old.py & cygnus_occlusion_new.py: these scripts were used for initial testing of generating occluded cygnus images. cygnus_occlusion_old.py generates labels with correct bboxes that go off the edge of the screen by cropping the final image after extracting the bbox from the mask. cygnus_occlusion_new.py uses the current technique for occlusion of achieving occlusion by setting offsets near the edge of the frame(included as an option in gen_cygnus_dataset.py).
- cygnus_keypoints.py: The base script for generating non-augmented cygnus images labeled with bboxes and keypoints. no augmentations are included in this script
- dynamic_moon.py: This script is used for generating images of gateway with dynamically sized moons, glare, blur, and domain-randomized-backgrounds
- SynImage_moon.py: This script was used to generate images of the moon from multiple distances and lighting angles used dynamicically-sized moon backgrounds
- cygnus_interpolated_keypoints.py: This script is used to generate non-augmented, interpolated image sequences of cygnus labeled with keypoints and bboxes