Я новичок в глубоком обучении и питоне, я пытался запустить keras R-FCN, я Google colab, и у меня возникла такая ошибка

tensorflow.python.framework.errors_impl.FailedPreconditionError: 2 root error(s) found.
  (0) Failed precondition:  Error while reading resource variable _AnonymousVar404 from Container: localhost. This could mean that the variable was uninitialized. Not found: Resource localhost/_AnonymousVar404/N10tensorflow3VarE does not exist.
     [[node regr_vote/crop_to_bounding_box_7/stack/ReadVariableOp_1 (defined at usr/local/lib/python3.6/dist-packages/keras/backend/tensorflow_backend.py:3009) ]]
  (1) Cancelled:  Function was cancelled before it was started
0 successful operations.
0 derived errors ignored. [Op:__inference_keras_scratch_graph_16906]

Я думаю, что у функции crop_to_bounding_box есть что-то, что не инициализировано, и это код, который называется crop_to_bounding_box

# position-sensitive ROI pooling + classify
        score_map_bins = []
        for channel_step in range(self.k*self.k):
            bin_x = K.variable(int(channel_step % self.k) *
                               self.pool_shape, dtype='int32')
            print(bin_x)
            bin_y = K.variable(int(channel_step / self.k) *
                               self.pool_shape, dtype='int32')
            channel_indices = K.variable(list(range(
                channel_step*self.channel_num, (channel_step+1)*self.channel_num)), dtype='int32')
            croped = tf.image.crop_to_bounding_box(
                tf.gather(pooled, indices=channel_indices, axis=-1), bin_y, bin_x, self.pool_shape, self.pool_shape)
            # [pool_shape, pool_shape, channel_num] ==> [1,1,channel_num] ==> [1, channel_num]
            croped_mean = K.pool2d(croped, (self.pool_shape, self.pool_shape), strides=(
                1, 1), padding='valid', data_format="channels_last", pool_mode='avg')
            # [batch * num_rois, 1,1,channel_num] ==> [batch * num_rois, 1, channel_num]
            croped_mean = K.squeeze(croped_mean, axis=1)
            score_map_bins.append(croped_mean)

Я использую tenorflow-GPU v2.1.0 и Keras 2.3.1

ОБНОВЛЕНО: вероятно, потому, что я добавляю K.variable внутри цикла, если я пробовал прокомментировать K.variable, код работал отлично, но мне нужно изменить K.variable на основе channel_step.

Как обновить K.variable, чтобы я мог определить K.variable вне цикла и обновить значение внутри цикла на основе channel_step?

2
Kevin Nathaniel 16 Апр 2020 в 15:51

1 ответ

Лучший ответ

Я нашел решение этой проблемы, мне просто нужно понизить версию Keras и TensorFlow. Сейчас я использую tenorflow-GPU 1.15.0 и Keras 2.2.4. Судя по всему, этот код не поддерживает TensorFlow 2 и новее.

0
Kevin Nathaniel 26 Апр 2020 в 02:31