用Caltech行人数据训练yolo2

因为想用yolo来做行人检测,所以尝试下载了caltech的行人检测数据集来fine-tuning。

首先去官网(http://www.vision.caltech.edu/Image_Datasets/CaltechPedestrians/)下载数据集

但是下载的数据,是seq格式,说明则为vbb格式,所以先要处理一下,变成yolo需要的jpg和txt。

用python写的将.seq 转为.jpg的代码, rootdir为.seq文件的文件夹,结果为每一帧的图像。

import os.path  
import fnmatch  
import shutil  
   
def open_save(file,savepath):  
    # read .seq file, and save the images into the savepath  
       
    f = open(file,'rb')  
    string = str(f.read())  
    splitstring = "\xFF\xD8\xFF\xE0\x00\x10\x4A\x46\x49\x46"  
    # split .seq file into segment with the image prefix  
    strlist=string.split(splitstring)  
    f.close()  
    count = 0  
    # delete the image folder path if it exists  
    if os.path.exists(savepath):  
        shutil.rmtree(savepath)  
    # create the image folder path  
    if not os.path.exists(savepath):  
     
  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 7
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 7
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值