Sunday 22 September 2019

Shapes (Circle, Triangle, Square, Star) dataset generator

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