python-argparse,read/write,os.path

argparse

import argparse
parser=argparse.ArgumentParser(description='test param')
parser.add_argument('--test_video_path',type=str)
parser.add_argument('--epoch',type=int)
parser.add_argument('--loss',type=float)
args=parser.parse_args()
test_video_path=args.test_video_path
epoch=args.epoch 
loss=args.loss

read/write

read

read code 1

for line in open(predict_file):
            line=line.split(' ')[2]
            data_list.append(line)

read code 2

with open(predict_file) as f:
            lines=f.readlines()
            for line in lines:
                line=line.rstrip()
                line=line.split(' ')[2]
                data_list.append(line)

read code 3

import yaml
with open('config.yml','r') as yamlfile:
    cfg=yaml.load(yamlfile)
img_W=cfg['img_W']
img_H=cfg['img_H']
time_length=cfg['time_length']
img_size=(img_W,img_H)

write

write code 1

import os
with open(os.path.join(test_result_path,'{}_predict.txt'.format(video_name)),'a') as f:
                f.write('{:04d} loss {:.6f}\n'.format(frame_num,frame_loss))

write code 2

import h5py
dataset_file=h5py.File('{}.h5'.format(style),'w')
create_dataset=dataset_file.create_dataset('data',(total_rows,time_length,img_W,img_H,1),maxshape=(None,time_length,img_W,img_H,1))
create_dataset[:,:]=h5_data
dataset_file.close()

os.path

create file folder

import os
if not os.path.exists(test_result_path):
        os.makedirs(test_result_path)

file list

file list 1

import os
import tqdm
for video_path in tqdm(os.listdir(test_video_path)):
        video=os.path.join(test_video_path,video_path)
        video_name=video.split('/')[-1].split('.')[0]
        print("######process {}".format(video))

file list 2

import os
import h5py
import tqdm
h5s_filelist=sorted([os.path.join(h5_store_path,item) for item in os.listdir(h5_store_path)])
dataset_path=os.path.join('/'.join(h5_store_path.split('/')[:-1]))
for h5_file_num,h5_file in enumerate(tqdm(h5s_filelist)):
        h5_data_file=h5py.File(h5_file,'r')
        h5_data=h5_data_file['data']