目标检测YOLOV2-训练集数据预处理（二）

目录

1. 加载必要的包

2. 预处理函数

3. 加载队列

4. 可视化

5. 测试

---

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. 预处理函数

主要是加载训练集路径后读取图片，解码，返回图片三通道数据信息