示例#1
0
SEARCH_ROOT = 'Z:\\Video'
REQUIED_SUBSTRING = '1080'

BATCH_SIZE = 2000

BUFSIZE = 10**5
SCALE_ALGO = 'bilinear'
PIXEL_FORMAT = 'rgb24'
RATE = 1
SAVE_DIR = 'ScannedFiles'

os.path.exists(SAVE_DIR) or os.mkdir(SAVE_DIR)

model = Network(weights='imagenet', include_top=True)
model.summary()
model = Model(inputs=model.input, outputs=model.get_layer("fc2").output)

MODLE_INPUT_DIMS = model.input.shape[1], model.input.shape[2]

fileList = []
for root, dirs, files in os.walk(SEARCH_ROOT):
    for f in files:
        type_guess = mimetypes.guess_type(f)
        filePath = os.path.join(root, f)
        if type_guess is not None and type_guess[
                0] is not None and 'video' in type_guess[
                    0] and REQUIED_SUBSTRING in filePath:

            fileList.append(filePath)
            print('Found', len(fileList), 'video files')