Skip to content

expected shape=(None, 299, 299, 3), found shape=(None, 224, 224, 3) but image is really (1, 299, 299, 3) #138

@wanghaifengnihaoa

Description

@wanghaifengnihaoa

use:nsfw-predict --saved_model_path ./nsfw.299x299.h5 --image_source images/色情1-2.jpeg

but:

Metal device set to: Apple M1

systemMemory: 16.00 GB
maxCacheSize: 5.33 GB

images/色情1-2.jpeg size: (224, 224)
Traceback (most recent call last):
  File "/Users/wanghaifeng/miniconda3/envs/tensorflow/bin/nsfw-predict", line 8, in <module>
    sys.exit(main())
  File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/nsfw_detector/predict.py", line 109, in main
    image_preds = classify(model, config['image_source'], config['image_dim'])
  File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/nsfw_detector/predict.py", line 64, in classify
    probs = classify_nd(model, images)
  File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/nsfw_detector/predict.py", line 71, in classify_nd
    model_preds = model.predict(nd_images)
  File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/utils/traceback_utils.py", line 70, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/var/folders/wn/r9w735dn2f958w3hhv5_2th00000gn/T/__autograph_generated_file_3gw2_vs.py", line 15, in tf__predict_function
    retval_ = ag__.converted_call(ag__.ld(step_function), (ag__.ld(self), ag__.ld(iterator)), None, fscope)
ValueError: in user code:

    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/engine/training.py", line 2341, in predict_function  *
        return step_function(self, iterator)
    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/engine/training.py", line 2327, in step_function  **
        outputs = model.distribute_strategy.run(run_step, args=(data,))
    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/engine/training.py", line 2315, in run_step  **
        outputs = model.predict_step(data)
    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/engine/training.py", line 2283, in predict_step
        return self(x, training=False)
    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/utils/traceback_utils.py", line 70, in error_handler
        raise e.with_traceback(filtered_tb) from None
    File "/Users/wanghaifeng/miniconda3/envs/tensorflow/lib/python3.10/site-packages/keras/src/engine/input_spec.py", line 298, in assert_input_compatibility
        raise ValueError(

    ValueError: Input 0 of layer "model_1" is incompatible with the layer: expected shape=(None, 299, 299, 3), found shape=(None, 224, 224, 3)

get shape:

import numpy as np
from PIL import Image

image = Image.open("images/色情1-2.jpeg")
print("image_size:", image.size)
image_array = np.array(image)
image_array = np.expand_dims(image_array, axis=0)
print("image_shape:", image_array.shape)

is:

image_size: (299, 299)
image_shape: (1, 299, 299, 3)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions