TypeError: неверный тип операнда для унарного -: 'Dimension' - PullRequest
0 голосов
/ 08 мая 2020

Я пытаюсь запустить свою модель на 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?

...