'''物体识别'''
importcv2 as cvimportosimportwarningsimportnumpy as npimporthmmlearn.hmm as hl
warnings.filterwarnings('ignore', category=DeprecationWarning)
np.seterr(all='ignore')defsearch_objects(directory):
directory=os.path.normpath(directory)if notos.path.isdir(directory):raise IOError('the directory' + directory + 'doesnt exist!')
objects={}for curdir, subdirs, files inos.walk(directory):for jpeg in (file for file in files if file.endswith('.jpg')):
path=os.path.join(curdir, jpeg)
label= path.split(os.path.sep)[-2]if label not inobjects:
objects[label]=[]
objects[label].append(path)returnobjects
train_objects= search_objects('./ml_data/objects/training/')print(train_objects)
train_x, train_y=[], []for label, filenames intrain_objects.item