前面介绍过在yolov5中训练自己的数据的数据集。EfficicentDet是google通过AutoML搞的单阶段目标检测模型,也是sota。所以本篇尝试用efficientdet进行训练。 准备数据 数据准备阶段与yolov5相同,参考上篇。尽量让数据准备阶段的代码和文件和复用。 此阶段生成了train.txt,valid.txt,text.txt 数据读入 数据读入主要是为了把适合模型输入的数据提供给训练模型。 此阶段也可以参照yolov5,大部分代码都相同。