kappazeta / km_predict

S2 full image prediction
Apache License 2.0
20 stars 9 forks source link

Failed to find data adapter that can handle input #12

Closed indrek-sunter closed 2 years ago

indrek-sunter commented 2 years ago

When run with TensorFlow 2.6, cm_predict throws the following error:

    2022-01-03 15:35:16.177467: W tensorflow/core/framework/cpu_allocator_impl.cc:80] Allocation of 37748736 exceeds 10% of free system memory.
    Traceback (most recent call last):
    File "/home/kappazeta/Documents/projects/cm_predict/cm_predict.py", line 390, in <module>
    main()
    File "/home/kappazeta/Documents/projects/cm_predict/cm_predict.py", line 385, in main
    cmf.predict()
    File "/home/kappazeta/Documents/projects/cm_predict/cm_predict.py", line 235, in predict
    predictions = self.model.predict(predict_generator)
    File "/home/kappazeta/Documents/projects/cm_predict/model.py", line 208, in predict
    preds = self.model.predict(dataset_pred)
    File "/home/kappazeta/micromamba/envs/cm_predict/lib/python3.9/site-packages/keras/engine/training.py", line 1720, in predict
    data_handler = data_adapter.get_data_handler(
    File "/home/kappazeta/micromamba/envs/cm_predict/lib/python3.9/site-packages/keras/engine/data_adapter.py", line 1383, in get_data_handler
    return DataHandler(*args, **kwargs)
    File "/home/kappazeta/micromamba/envs/cm_predict/lib/python3.9/site-packages/keras/engine/data_adapter.py", line 1137, in __init__
    adapter_cls = select_data_adapter(x, y)
    File "/home/kappazeta/micromamba/envs/cm_predict/lib/python3.9/site-packages/keras/engine/data_adapter.py", line 976, in select_data_adapter
    raise ValueError(
    ValueError: Failed to find data adapter that can handle input: <class 'data_generator.DataGenerator'>, <class 'NoneType'>

It seems to be working better with TensorFlow 2.4. The predictor would eventually need to be ported to TF 2.6.