Используйте resnet50 для регрессии - керас

Question

Я играю с обучением управлению автомобилем в видеоигре.Я подхожу к этому как к проблеме регрессии с выходом для значения рулевого управления между -1 и 1.

После некоторых исследований кажется, что модель resnet50 может быть хорошим местом для начала ... Но чтениев документации по keras указано количество классов.

Можно ли использовать ее для регрессии?

Или я должен подойти к нему как к проблеме классификации и разбить диапазон рулевого управления на несколько классов?

Answer 1

Вы можете использовать его для регрессии, модели обнаружения, такие как Faster R-CNN, SSD, YOLO, используют регрессию для прогнозирующей части ограничительной рамки.В таком случае вы обычно заменяете перекрестную энтропийную потерю на среднеквадратическую потерю.Если диапазон выходных значений не слишком сильно отличается, вы также можете использовать классификационный подход.Если количество классов не окажется слишком большим, это может на самом деле работать лучше.