Дрессировкой занялась нейросеть

В США создан аппарат для полуавтоматической дрессировки собак. Гаджет использует для распознавания положения тела собаки нейросеть: если команда выполнена правильно, собака получает лакомство. Статья опубликована на arXiv.org.

Прототип аппарата для дрессировки

Джейсон Сток (Jason Stock) и Том Кейви (Tom Cavey) из Университета штата Колорадо решили использовать положительное подкрепление. Они посчитали, что дрессировать собак очень просто: нужно лишь подкреплять правильное поведение животного. После таких умозаключений ученые разработали алгоритм, адаптированный для работы на недорогом одноплатном компьютере.

Исследователи построили также прототип аппарата для дрессировки – устройство выдает лакомство, когда поступает соответствующий сигнал от компьютера. Фактически это отсек с кормом и дозатор, выдающий лакомство.

Разработчики использовали три распространенных команды: «сидеть», «стоять» и «лежать». Для их распознавания они опробовали различные модификации трех сверточных нейросетей: ConvNet, MobileNetV2 и EfficientNet, а после обучения опробовали квантование («сжатие» нейросети за счет преобразования чисел с плавающей запятой в целые, подробнее можно прочитать тут) с помощью TensorFlow и TensorRT.