Ищем реализацию реснета в тензорном потоке

Существуют ли какие-либо реализации реснета в тензорном потоке? Я наткнулся на несколько (например, https://github.com/ry/tensorflow-resnet, https://github.com/xuyuwei/resnet-tf), но эти реализации имеют некоторые ошибки ( например, см. раздел «Проблемы» на соответствующей странице github). Я хочу обучить imagenet с помощью resnet и ищу реализации tensorflow.


person mac_i    schedule 15.10.2016    source источник
comment
Вот реализация, которую я создал с помощью графического процессора TensorFlow в Docker, используя полный набор данных ImageNet, обученный на 2x NVIDIA RTX 2080 Ti. блог .exxactcorp.com/ Он в основном основан на официальной реализации TensorFlow с некоторыми советами и рекомендациями, которые я нашел в процессе. Также рассматривается запуск и взаимодействие с докерной средой TF. Дайте мне знать, если у вас есть какие-либо проблемы или вопросы! Джеймс   -  person James Montantes    schedule 07.04.2019
comment
Я хотел выяснить, есть ли лучший способ загрузить набор данных ImageNet. Я не мог найти прямого обходного пути для этого.   -  person prb_cm    schedule 29.07.2020


Ответы (3)


Есть несколько (50/101/152) в tensorflow:models/slim .

пример блокнота показывает, как получить предварительно обученный inception бег, res-net, вероятно, ничем не отличается.

person mdaoust    schedule 15.10.2016
comment
Привет~ Я обнаружил, что по предложенной вами ссылке есть модели resnet-v1 и v2. И я заметил, что для resnet-v1 есть предварительно обученные веса, но нет предварительно обученных весов для resnet-v2. Вы знаете, доступны ли сейчас предварительно обученные веса v2? - person LI Xuhong; 14.11.2016

Я реализовал версию ResNet для cifar10 с тензорным потоком. Ошибки проверки ResNet-32, ResNet-56 и ResNet-110 составляют 6,7%, 6,5% и 6,2% соответственно. (Вы можете легко изменить количество слоев в качестве гиперпараметров.)

Я старался быть дружелюбным с новым поклонником ResNet и написал все прямо. Вы можете запустить файл cifar10_train.py напрямую, без каких-либо загрузок.

https://github.com/wenxinxu/resnet_in_tensorflow

person Alex Wenxin Xu    schedule 05.12.2016

Я реализовал Resnet с помощью ronnie.ai и keras. Оба инструмента великолепны. В то время как Ронни легче с нуля.

person Master C    schedule 02.05.2018