Add another SDF representation like the sphere and the box

[Komti13]

Thank you for this solution, it is really helpful.
I want to add some other shapes to detect by the neural network but found that it is a little bit complicated
I would really appreciate if you give me some hints to begin with to add my own SDF shape
Thanks!