Как правильно настроить модель набора данных с разными классами?

Question

В этой статье автор точно настроил модель COCO с предварительной подготовкой на набор данных PoseTrack. Я хочу сделать что-то похожее, но я не уверен, как это сделать, поскольку количество классов (сочетаний клавиш) не одинаково.

Я думал об «отключении» определенных классов, но не уверен, как.

Answer 1

Вам нужно будет заменить последний слой nn числом классов вашей задачи и заново подготовить набор данных (предположим, что он имеет ту же форму ввода). Вы можете использовать предварительно обученные веса предыдущего слоя, чтобы сходимость была быстрее и вам потребовалось меньше данных