Darkflow: используя пример ошибки graph-yolo.pb для Android

Отсутствие Opkernel означает отсутствие реализации для оборудования, на котором работает этот .pb .
Чтобы решить эту проблему, посмотрите на класс reorg из ./net/ops/convolution.py . Он имеет два метода _forward и forward . Текущий вариант по умолчанию использует forward , который имеет extract_image_patches — встроенный метод тензорного потока.

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

я изменил код, но не могу сгенерировать .pb
/usr/bin/python3 ./flow.py --model /home/qkj/projects/dark_flow/darkflow/cfg/yolo-voc.cfg --load /home/qkj/projects/dark_flow/darkflow/bin/yolo- voc.weights --savepb

Traceback (последний последний вызов):
Файл "./flow.py", строка 42, в
tfnet = TFNet(ФЛАГИ)
Файл "/home/qkj/projects/dark_flow/darkflow/net/build.py", строка 50, в __init__
self.build_forward()
Файл "/home/qkj/projects/dark_flow/darkflow/net/build.py", строка 70, в build_forward
состояние = op_create(*аргументы)
Файл "/home/qkj/projects/dark_flow/darkflow/net/ops/__init__.py", строка 27, в op_create
вернуть op_types layer_type
Файл "/home/qkj/projects/dark_flow/darkflow/net/ops/baseop.py", строка 42, в __init__
self.forward ()
Файл "/home/qkj/projects/dark_flow/darkflow/net/ops/convolution.py", строка 13, вперед
для i в диапазоне (ч/с):
TypeError: объект 'float' не может быть интерпретирован как целое число

реорганизация класса (BaseOp):
защита вперед (я):
вход = self.inp.out
форма = inp.get_shape().as_list()
_, ч, ш, с = форма
s = self.lay.stride
выход = список()
для i в диапазоне (ч/с):
row_i = список()
для j в диапазоне (w/s):
si, sj = s * i, s * j
boxij = inp[:, si: si+s, sj: sj+s,:]
flatij = tf.reshape(boxij, [-1,1,1,c s s])
row_i += [flatij]
выход += [tf.concat(2, row_i)]
self.out = tf.concat (1, выход)

def _forward(self):
    inp = self.inp.out
    s = self.lay.stride
    self.out = tf.extract_image_patches(
        inp, [1,s,s,1], [1,s,s,1], [1,1,1,1], 'VALID')

Эй, это ошибка при переводе с Python2 на Python3. Спасибо за указание на это, я обновил код.

У меня такая же проблема, и я поменял местами имена двух функций, но файл yolo.pb по-прежнему вызывает ошибку в Android. Пожалуйста, помогите мне это исправить!!!

[[Узел: ExtractImagePatches = ExtractImagePatches[T=DT_FLOAT, ksizes=[1, 2, 2, 1], padding="VALID", rate=[1, 1, 1, 1], strides=[1, 2, 2 , 1]] (47-дырявый)]]

Спасибо