dbnet代码运行要点

本文档介绍了如何运行 DBNet 模型,包括数据集预处理、配置文件修改、直接运行 Python 文件以及解决常见错误。确保数据集的正确格式,修改配置文件中的数据路径和基路径至绝对路径,注意Python版本和torchvision版本的兼容性。
摘要由CSDN通过智能技术生成

1. 关于数据集预处理

把训练数据train和测试数据test的img和gt,放到datasets文件夹下,或者其他文件夹中也可以,代码中写出数据集所在位置的绝对路径即可。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
将训练数据和测试数据生成如下图的格式:
参考文献
在这里插入图片描述
生成train.txt和test.txt,保存到datasets文件夹下(必须也把test.txt也生成)。
生成文件的代码如下:
在这里插入图片描述

import os
def get_images(img_path):
    '''
    find image files in data path
    :return: list of files found
    '''
    img_path = os.path.abspath(img_path)
    files = []
    exts = ['jpg', 'png', 'jpeg', 'JPG', 'PNG']
    for parent
  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值