-
Notifications
You must be signed in to change notification settings - Fork 294
Open
Description
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
Labels
No labels