0

警告(来自警告模块):文件“C:\Users\Pc\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\keras\engine\training.py”,第 1844 行 warnings.warn ('Model.fit_generator已弃用且 ' UserWarning:Model.fit_generator已弃用并将在未来版本中删除。请使用Model.fit支持生成器的 .Epoch 1/20 Traceback (最近调用最后一次): File "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 行,在调用中 过滤的平面参数,捕获的输入=图形函数。捕获的输入) 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 行,在call ctx=ctx) 文件“C:\Users\Pc\AppData\Local\Programs\Python\Python37\lib\site-packages\tensorflow\python\eager\execute.py”,第 60 行,在 quick_execute 输入,attrs, 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 import ClassificationModelTrainer

model_trainer = 分类模型训练器()

model_trainer.setModelTypeAsMobileNetV2()

model_trainer.setDataDirectory(r"C:\Users\Pc\Desktop\model\minioni")

model_trainer.trainModel(num_objects=500,num_experiments=1,enhanced_data=True,batch_size=32,show_network_summary=True)

4

0 回答 0