机器学习实战:用efficientdet训练自己的数据集

前面介绍过在yolov5中训练自己的数据的数据集。EfficicentDet是google通过AutoML搞的单阶段目标检测模型,也是sota。所以本篇尝试用efficientdet进行训练。

准备数据

数据准备阶段与yolov5相同,参考上篇。尽量让数据准备阶段的代码和文件和复用。
此阶段生成了train.txt,valid.txt,text.txt

数据读入

数据读入主要是为了把适合模型输入的数据提供给训练模型。
此阶段也可以参照yolov5,大部分代码都相同。

你可能感兴趣的:(机器学习,深度学习,人工智能)