基于tensorflow2的yolov4训练自己的数据集(三)

基于tensorflow2.0的yolov4训练自己的数据集（三）

      在windows和linux分别训练了基于tensorflow2的yolov4模型，将方法记录下来，方便其他同学参考借鉴。内容较多，分批写出来。上一次写到生成了基于darknet的yolov4模型：上一集，继续往下进行。

一、将darknet模型转化为tensorflow模型

在官网提供的github项目，直接可以将darknet转化为tensorflow的pb模型，转化地址。

下载解压后目录如下图所示：

修改配置文件：

进入到core文件夹，找到config.py文件，这里面是转化、训练、测试模型的一些配置参数，如下图所示：

其他的可以暂时不用理会，打开config.py文件夹，如下所示：

#! /usr/bin/env python
# coding&#