目录
1. 加载必要的包
文件名:get_dataset.py
本文件只是整个YOLO项目中的一个,要加载一个xml_parase.py文件
# -*- coding: utf-8 -*-
from xml_parse import paras_annotation
import os
import matplotlib.pyplot as plt
from matplotlib import patches
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2'
import tensorflow as tf
TensorFlow版本2.0.0及以上
2. 预处理函数
主要是加载训练集路径后读取图片,解码,返回图片三通道数据信息