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']