<计算机视觉二> labelme标定的数据转换成yolo训练格式
上章节讲到了如何将要数据模型转换成网络训练的时候方便读取的数据格式,本文讲解下pytorch模块如何使用这个转换后的数据代码块如下:
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
'''
@File : datasets.py
@Time : 2021/08/19 16:37:10
@Author : XIA Yan
@Contact : 2440212215@qq.com
@Version : 0.1
@License : Apache License Version 2.0, January 2004
@Language: pytho