微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

我正在尝试使用 imageai 训练模型,但出现以下错误:

如何解决我正在尝试使用 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]

函数调用栈: train_function

我正在学习本教程: 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 举报,一经查实,本站将立刻删除。