如何解决我正在尝试使用 imageai 训练模型,但出现以下错误:
警告(来自警告模块):
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\keras\engine\training.py”,第 1844 行
warnings.warn('Model.fit_generator
已弃用,'
用户警告:Model.fit_generator
已弃用,将在未来版本中删除。请使用支持生成器的 Model.fit
。
纪元1/20
回溯(最近一次调用最后一次):
文件“C:\Users\Pc\Desktop\model\trainn.py”,第 9 行,在
model_trainer.trainModel(num_objects=500,num_experiments=20,enhance_data=True,batch_size=32,show_network_summary=True)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\imageai\Classification\Custom_init_.py”,第 395 行,在 trainModel 中
validation_steps=int(num_test / batch_size),callbacks=[checkpoint,lr_scheduler])
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\keras\engine\training.py”,第 1861 行,在 fit_generator
initial_epoch=initial_epoch)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\keras\engine\training.py”,第 1100 行,适合
tmp_logs = self.train_function(iterator)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\def_function.py”,第 828 行,调用
结果 = self._call(*args,**kwds)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\def_function.py”,第 888 行,在 _call
返回 self._stateless_fn(*args,**kwds)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\function.py”,第 2943 行,调用
Filtered_flat_args,capture_inputs=graph_function.captured_inputs) # pylint: disable=protected-access
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\function.py”,第 1919 行,call_flat
ctx、args、cancellation_manager=cancellation_manager))
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\function.py”,第 560 行,调用中
ctx=ctx)
文件“C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\tensorflow\python\eager\execute.py”,第 60 行,在 quick_execute 中
输入、属性、num_outputs)
tensorflow.python.framework.errors_impl.InvalidArgumentError:logits 和标签必须是可广播的:logits_size=[32,500] labels_size=[32,2]
[[node categorical_crossentropy/softmax_cross_entropy_with_logits(定义在 C:\Users\Pc\AppData\Local\Programs\Python\python37\lib\site-packages\imageai\Classification\Custom_init. py:395) ]] [Op:__inference_train_function_9811]
我正在学习本教程: https://imageai.readthedocs.io/en/latest/custom/
这是我的代码: from imageai.Classification.Custom 导入ClassificationModelTrainer
model_trainer = ClassificationModelTrainer()
model_trainer.setModelTypeAsMobileNetV2()
model_trainer.setDataDirectory(r"C:\Users\Pc\Desktop\model\minioni")
model_trainer.trainModel(num_objects=500,num_experiments=1,enhance_data=True,batch_size=32,show_network_summary=True)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。