argparser

把train.txt作为VOCDataSet的参数txtname传入

VOCDateSet构造函数里文件路径

调用parse_xml_to_dict将每个xml文件的object提取放入data内

将从data里提取出的object信息放入boxes和labels中

修改split_data内容分离出json_train和json_val文件

构造函数

python之parser.add_argument()用法——命令行选项、参数和子命令解析器_夏普通的博客-CSDN博客_parser.add_argument

import

argparse.ArgumentParser(description__doc__)#

default3,

helpdevice)parser.add_argument(--B,

default4,

parser.parse_args()print(args.A)print(args.A)print(type(args.A))print(type(args.B))print(int(args.A)1)print(int(args.A)1)

在Edit

python中assert的用法简洁明了_花里梦雨的博客-CSDN博客_python

assert用法

lxml.etree.fromstring的使用_夏夏今天学习了吗的博客-CSDN博客

对象Element

Python常用小技巧五——批量读取json文件_码农邦的博客-CSDN博客_python批量读取json文件

正脸红外光

修改mydataset文件将json转数据集能达到这个效果就o

主文件

把train.txt作为VOCDataSet的参数txtname传入

VOCDateSet构造函数里文件路径

调用parse_xml_to_dict将每个xml文件的object提取放入data内

data

self.parse_xml_to_dict(xml)[annotation]print(--------------data-------------------)print(type(data))print(data)

调用函数后data

--------------data-------------------

class

}将从data里提取出的object信息放入boxes和labels中

for

进一步检查数据有的标注信息中可能有w或h为0的情况这样的数据会导致计算回归loss为nanif

xmax

0.format(xml_path))continueboxes.append([xmin,

ymin,

ymax])labels.append(self.class_dict[obj[name]])if

difficult

obj:iscrowd.append(int(obj[difficult]))else:iscrowd.append(0)

本来这个文件存储的就是类别对应的序号这里是在给自制的数据集可视化的部分。

修改split_data内容分离出json_train和json_val文件

构造函数

data[shapes]:#用多边形标注时#print(obj)if

obj[shape_type]polygon:#print(len(obj[points]))label

obj[label]xminymin100000xmaxymax0for

point

point[0]xmin:xminfloat(point[0])elif

point[0]xmax:xmaxfloat(point[0])if

point[1]ymin:yminfloat(point[1])elif

point[1]ymax:ymaxfloat(point[1])else:label

obj[label]xmin

在多边形的处理中找出点x的最小最大值和y的最小最大值分别赋值给xmin

xmax