SSD
weixin_43730508
每天进步一点点
展开
-
目标检测性能评价指标(mAP、IOU..)
参考这位博主的解答解释的很详细转载 2020-09-03 10:57:29 · 568 阅读 · 0 评论 -
python修改图片的大小
'''import osfrom PIL import Imagedef save_resize_img(img_path, img_path_train, width, height): if not os.path.exists(img_path_train): os.mkdir(img_path_train) number = 0 for label_name in os.listdir(img_path): if not os.pat原创 2020-06-11 19:04:38 · 2933 阅读 · 1 评论 -
修改xml的属性值
# coding=utf-8import osimport os.pathimport xml.dom.minidom#更改xml文件中name和folder属性值path = "F:/mx_matting/ssd.pytorch-master/data/VOCdevkit/VOC2007/Annotations"files = os.listdir(path) # 得到文件夹下所有文件名称for xmlFile in files: # 遍历文件夹 if not os.path原创 2020-06-11 18:58:15 · 898 阅读 · 0 评论 -
python按照比例分配数据集:train、test、val
#把所有的xml文件分类为:val train test trainval#比例是:6:2:2import osimport randomtrainval_percent = 0.8train_percent = 0.6xmlfilepath = 'Annotations' #Annotations地址txtsavepath = 'ImageSets/Main' #Main地址total_xml = os.listdir(xmlfilepath)total_xml.sort()prin原创 2020-05-20 11:33:12 · 4057 阅读 · 0 评论 -
win10下用SSD-pytorch训练自己的数据集
代码链接https://github.com/amdegroot/ssd.pytorch权重下载地址:链接:https://pan.baidu.com/s/1EqrkQOR0Vx4JGaqJn0um8w提取码:rd8c环境配置:我是用的yolov3-pytorch的环境,不过一定要注意:pytorch版本必须小于1.3,我用的是pytorch1.2.0修改SSD代码以适应自己的数据集参考这位博主对应自己的数据集进行修改,我的数据集是一分类。有一点需要注意:如果和我一样是一分类,要在VOC原创 2020-05-19 23:30:38 · 2873 阅读 · 7 评论