load .mat data with python

import scipy.io as sio

import numpy as np

import h5py

import os


IMG_PATH= '../../data/my_bus/images/'

SAVE_PATH = '../../data/my_bus/'

ANNOT_PATH = '../../data/my_bus/Annotations'


if not os.path.exists(SAVE_PATH):

        os.mkdir(SAVE_PATH)


annot_list = os.listdir(ANNOT_PATH)

img_list = os.listdir(IMG_PATH)


for i in range(0, len(annot_list)):

    path = os.path.join(ANNOT_PATH, annot_list[i])

    print path

for filename in annot_list:

    portion = os.path.splitext(filename)

#  print portion[0]

#  print portion[1]

    annot_file = ANNOT_PATH + '/' + filename

    try:

        data = sio.loadmat(annot_file)

    except:

        print 'pass', filename

        continue

    record = data['record']

    size = record['size']

    objects = record['objects']

猜你喜欢

转载自blog.csdn.net/doublechenchenchen/article/details/80589786