search

ria-com / nomeroff-net

Nomeroff Net. Automatic numberplate recognition system.
GNU General Public License v3.0
459 stars 159 forks source link

Проблема с tensorflow 2.2.0 #91

Closed pashey173 closed 4 years ago

pashey173 commented 4 years ago

Здравствуйте. Пытаюсь запустить Ваш код на tensorflow 2.2.0, вылетает ошибка:

AttributeError: module 'tensorflow' has no attribute 'ConfigProto'

Возможно ли каким - то образом запустить demo3 на версии 2.2.0?

И как запустить файл demo3 с использованием GPU? Спасибо.

ApelSYN commented 4 years ago

Там будем множество несовместимх строчек. Если нет возможности поставить tensorflow 1.5.x, запускайте в docker-е

pashey173 commented 4 years ago

Я уже поставил 1.15.2 Запускается на CPU, но если пытаюсь запустить на GPU, то выходит ошибка

Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node conv1/convolution}}]] [[mrcnn_detection/Reshape_1/_5635]]

Хотелось бы запустить все - таки на GPU. Проверил работоспособность GPU с NVIDIA и CUDA Везде True

ApelSYN commented 4 years ago

Tensorflow может быть собран под устаревшуюю cuda. Поробуйте сами собрать tf по инструкции https://www.tensorflow.org/install/source

сб, 25 лип. 2020, 15:32 користувач pashey173 notifications@github.com пише:

Я уже поставил 1.15.2 Запускается на CPU, но если пытаюсь запустить на GPU, то выходит ошибка

Unknown: Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [[{{node conv1/convolution}}]] [[mrcnn_detection/Reshape_1/_5635]]

Хотелось бы запустить все - таки на GPU. Проверил работоспособность GPU с NVIDIA и CUDA Везде True

— You are receiving this because you commented. Reply to this email directly, view it on GitHub https://github.com/ria-com/nomeroff-net/issues/91#issuecomment-663850384, or unsubscribe https://github.com/notifications/unsubscribe-auth/AAFTPVX4YM5ZKQ54FI6FQ63R5LGFRANCNFSM4PGWBTQQ .

pashey173 commented 4 years ago

Нужно что - то добавлять в коде, что бы ускорить процесс загрузки модели и т.д. с использованием GPU?

ApelSYN commented 4 years ago

В память GPU модель грузится немного дольше чем в ОЗУ, ускорение происходит на этапе инференса. Ускорить работу можно написав обертку в виде сервиса. Посмотрите другие issues, там уже про это спрашивали и отвечали неоднократно https://github.com/ria-com/nomeroff-net/issues/84

ApelSYN commented 4 years ago

Уже есть бета для TensorFlow 2.x без Mask RCNN, можете пробовать: https://github.com/ria-com/nomeroff-net/tree/v1.0

ApelSYN commented 4 years ago

Уточнение -- модели натренированы для tensorflow >=2.3