/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.

Primary LanguageJupyter NotebookGNU General Public License v3.0GPL-3.0

Stargazers