Я пытаюсь запустить свою модель на Jetson Xavier, но получаю следующее сообщение об ошибке:
TypeError Traceback (most recent call last)
<ipython-input-22-22297f98b294> in <module>
----> 1 footpath_model = FootpathModelArchitecture(input_shape).footpath_model
~/footpath_segmentation/src/architecture.py in __init__(self, input_shape)
100 row_shape_diff = self.upscore32.get_shape()[1] - self.inputs.get_shape()[1]
101 column_shape_diff = self.upscore32.get_shape()[2] - self.inputs.get_shape()[2]
--> 102 self.upscore32 = Cropping2D(cropping=((0, row_shape_diff), (0, column_shape_diff)))(self.upscore32)
103
104 self.final_conv = Conv2D(filters=1, kernel_size=(1, 1), strides=(1, 1),
/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/keras/engine/base_layer.py in __call__(self, inputs, *args, **kwargs)
776 outputs = base_layer_utils.mark_as_return(outputs, acd)
777 else:
--> 778 outputs = call_fn(cast_inputs, *args, **kwargs)
779
780 except errors.OperatorNotAllowedInGraphError as e:
/usr/local/lib/python3.6/dist-packages/tensorflow_core/python/keras/layers/convolutional.py in call(self, inputs)
2522 -self.cropping[1][1], :]
2523 elif self.cropping[1][1] == 0:
-> 2524 return inputs[:, self.cropping[0][0]:-self.cropping[0][1],
2525 self.cropping[1][0]:, :]
2526 return inputs[:, self.cropping[0][0]:-self.cropping[0][1], self.cropping[
TypeError: bad operand type for unary -: 'Dimension'
Версия tenorflow на Jetson Xavier: tensorflow == 2.1.0 + nv20 .4 . На моем основном компьютере это tensorflow == 2.1.0rc2 , и там он выполняется правильно и без проблем. Я также добился успеха на машинах с tensorflow == 2.1.0 . Может быть, проблема в версии tenorflow для Xavier?