Python 对象检测模型_main.py错误:无法批处理组件1中具有不同形状的张量
我正在尝试使model_main.py文件适用于我正在创建的自定义对象检测器。然而,我似乎遇到了以下错误,并且不知道如何修复它。任何帮助都将不胜感激 我正在运行以下命令:Python 对象检测模型_main.py错误:无法批处理组件1中具有不同形状的张量,python,python-3.x,tensorflow,machine-learning,object-detection,Python,Python 3.x,Tensorflow,Machine Learning,Object Detection,我正在尝试使model_main.py文件适用于我正在创建的自定义对象检测器。然而,我似乎遇到了以下错误,并且不知道如何修复它。任何帮助都将不胜感激 我正在运行以下命令: sudo python3 /data/tensorflow/models/research/object_detection/model_main.py --pipeline_config_path=/data/tensorflow/models/research/object_detection/samples/config
sudo python3 /data/tensorflow/models/research/object_detection/model_main.py --pipeline_config_path=/data/tensorflow/models/research/object_detection/samples/configs/faster_rcnn_resnet50_coco.config —model_dir=/home/USER/TASK/models/faster_rcnn_resnet50_coco/ --num_train_steps=200000 --num_eval_steps=1500 --sample_1_of_n_eval_examples=1 --alsologtostderr
错误如下:
InvalidArgumentError (see above for traceback): Cannot batch tensors with different shapes in component 1. First element had shape [600,900,3] and element 1 had shape [512,1024,3].
完整的回溯在
我认为问题在于配置文件中的图像大小调整器。可能我的图像太大,导致洗牌缓冲区内存不足?但是,我无法通过更改“最小”和“最大调整大小”值来解决此问题
编辑:经过更多的尝试,我不认为这是缓冲区大小的问题。我将其进一步降低到512MB,填充良好,但仍然收到相同的错误,请参见:您尝试过填充吗?不确定,为什么以前的版本没有它就可以工作
keep_aspect_ratio_resizer {
pad_to_max_dimension: true
min_dimension: 600
max_dimension: 1024
}
看来我也一样。这能解决吗?有解决办法吗?我很遗憾地无法解决这个问题。如果你运气好,请告诉我!