utils.py 文件源码

def get_file_list(folder):
    names = os.listdir(folder)
    fnames = []
    bboxes = []
    bbox_names = map(lambda name: os.path.join(
        folder, name, '_bboxes.txt'), names)
    with Parallel(n_jobs=-1) as parallel:
        dfs = parallel(delayed(_extract_names_bboxes)(bname)
                       for bname in bbox_names)
    df = pd.concat(dfs, ignore_index=True)
    df['Flag'] = df['Name'].map(lambda x: True if os.path.exists(x) else False)
    print "Initial number of images:", df['Name'].count()
    df = df[df['Flag'] == True]
    print "Total number of existing images:", df['Name'].count()
    return df['Name'].values, df['BBox'].values