fvilmos / multi_task_learning_shape_color_position

Implementation of a slim multi-task learner DNN with 3 tasks (heads). Detects shape type, color, the position of the midpoint of the shape.

Geek Repo

Github PK Tool