Конвертируйте модель, чтобы использовать меньшее входное изображение с возможной базелью - PullRequest
Новая
       16

Конвертируйте модель, чтобы использовать меньшее входное изображение с возможной базелью

0 голосов
/ 27 апреля 2019

Я обнаружил, что для работы с tenorflow mobile с ssd_mobilenet_v2 время вывода составляет около 320 мс (я не считаю, что время, используемое для масштабирования изображения до размера 200x200px).Я хотел бы уменьшить время вывода, используя изображение 50x50px вместо изображения 200x200px.К счастью, у меня есть несколько изображений в этом формате 50x50px.Поэтому я подумал, что было бы полезно преобразовать модель, используя bazel, чтобы она принимала входное изображение фиксированного меньшего размера и, таким образом, уменьшала время вывода.Знаете ли вы, можно ли вообще сказать предварительно обученной модели принимать изображения с более низким разрешением, для которых она, вероятно, была подготовлена?Будет ли модель все еще хорошо классифицировать и обнаруживать людей / объекты?

Я уже читал этот пост Tensorflow Lite: о входной форме в файле tflite .Поэтому я думаю, что можно конвертировать модели, использующие 

./bazel-bin/tensorflow/contrib/lite/toco/toco \
--input_file=/home/hwh/pnet.pb \
--input_format=TENSORFLOW_GRAPHDEF --output_format=TFLITE \ 
--output_file=/home/hwh/pnet.tflite --inference_type=FLOAT \ 
--input_type=FLOAT --input_arrays=pnet/input \
--output_arrays=pnet/prob1 --input_shapes=1,299,299,3

, в указанный размер входного изображения.Я не мог выяснить, рекомендуется ли использовать изображения меньшего размера, чем те, с которыми они тренировались, или мне нужно самостоятельно обучать всю модель, чтобы иметь возможность использовать меньшие входные изображения.Я буду использовать модель в этом приложении для Android .

Более того, я нашел модельный конфиг модели в модели zoo .Можно ли уменьшить размер формы (используя соотношение 1: 1, например, 50x50px) здесь, как здесь Модель API-интерфейса обнаружения Tensorflow с использованием 'keep_aspect_ratio_resizer' и здесь Размер входного изображения SSD Mobilenet ?Может ли это решить мою проблему вообще?

Я хотел бы уменьшить время вывода, используя меньшее входное изображение для предварительно обученной модели ssd_mobilenet_v2, и хотел бы знать, возможно ли это или необходимо обучать собственныйсама модель, которая имеет функциональность ssd_mobilenet?

Большое спасибо за помощь!

...