使用Efficientdet的Pytorch官方源码训练自己的数据集

只只只只只

已于 2022-08-13 22:23:47 修改

阅读量3.1k

点赞数 5

分类专栏：训练自己的数据集文章标签： pytorch python 深度学习

于 2021-09-08 15:28:52 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_45817307/article/details/120180273

版权

本文介绍了如何使用Efficientdet的Pytorch实现训练自己的数据集，包括从官网获取源码、数据集格式转换、配置yaml文件、修改类别、加载权重以及解决训练过程中的错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用Efficientdet的Pytorch官方源码训练自己的数据集

1.先从官网下载代码efficientdet源码

2.再将自己的数据集都放入datasets中。我自己的数据集格式是Pascal VOC的xml格式，这个代码需要的格式是json格式，所以在网上找了好久找到了一个适合的转换代码，这里我放上来

    # coding=utf-8
import xml.etree.ElementTree as ET
import os
import json


voc_clses = ['D00', 'D01', 'D10', 'D11', 'D20', 'D40', 'D43', 'D44', 'D50']
# 这里换成自己数据集的类别....


categories = []
for iind, cat in enumerate(voc_clses):
    cate = {
   }
    cate['supercategory'] = cat
    cate['name'] = cat
    cate['id'] = iind
    categories.append(cate)

def getimages(xmlname, id):
    sig_xml_box = []
    tree = ET.parse(xmlname)
    root = tree.getroot()
    images = {
   }
    for i in root:  # 遍历一级节点
        if i.tag == 'filename':
            file_name = i.text  # 0001.jpg
            # print('image name: ', file_name)
            images['file_name'] = file_name
        if i.tag == 'size':
            for j in i:
                if j.tag == 'width':
                    width = j.text
                    images['width'] = width
                if j.tag == 'height':
                    height = j.text
                    images['height'] = height
        if i.tag == 'object':
            for j in i:
                if j.tag == 'name':
                    cls_name = j.text
                cat_id = voc_clses.index(cls_name) + 1
                if j.tag == 'bndbox':
                    bbox = []
                    xmin = 0
                    ymin = 0
                    xmax = 0
                    ymax = 0
                    for r in j:
                        if r.tag == 'xmin':
                            xmin = eval(r.text)
                        if r.tag == 'ymin':
                            ymin = eval(r.text)
                        if r.tag == 'xmax':
                            xmax = eval(r.text)
                        if r.tag == 'ymax':
                            ymax = eval(r.text)
                    bbox.append(xmin)
                    bbox.append(ymin)
                    bbox.append(xmax - xmin)
                    bbox.append(ymax - ymin)
                    bbox.append(id)   # 保存当前box对应的image_id
                    bbox.append(cat_id)
                    # anno area
                    bbox.append((xmax - xmin) * (ymax - ymin

最低0.47元/天解锁文章