print(keras.__version__)
print(tf.__version__)
2.3.1
2.0.0
https://www.kaggle.com/super13579/u-net-base-on-resnet34-transfer-learning-keras
с использованием следующего кода
def pred_encode(img, **kwargs):
cur_seg, _ = predict(img)
cur_rles = multi_rle_encode(cur_seg, **kwargs)
return [[img, rle] for rle in cur_rles if rle is not None]
out_pred_rows = []
for c_img_name in tqdm_notebook(test_paths[:30000]): ## only a subset as it takes too long to run
out_pred_rows += pred_encode(c_img_name, min_max_threshold=1.0)
Но здесь я получил следующую ошибку
InvalidArgumentError: Размер 1 должен быть неотрицательным, а не -21 [[{{node u-resnet34 / bn_data / batchnorm / mul-1-ReshapeNHWCToNCHW-LayoutOptimizer}}]] [Op: __inference_keras_sc144949] ]
Стек вызовов функций: keras_scratch_graph