I have created a python script to generate shapes , for deep learning.
The code is on gitlab
here.
It can:
Generate circle, triangle, square and or star
Single or multiple shapes per image
1 or 3 color channels
Can define color range for each of shape as well as background
Can further use gradient color patterns
Can generate full shapes or outline
Supports rotation
Can set size scales
Supports annotation for darknet/yolo and Pascal VOC formats
Set output path
Here an example with Pascal VOC annotation:
Left: generated image, Center: Pascal VOC objects, Right: Pascal VOC classes
No comments:
Post a Comment