Closed Yii-maker closed 2 years ago
parser = argparse.ArgumentParser(description="extract optical flows")
parser.add_argument("--src_dir", default='./dataset/shanghaitech/videos/training')
parser.add_argument("--out_dir", default='./dataset/shanghaitech/dense_flow/training')
parser.add_argument("--num_worker", type=int, default=32)
parser.add_argument("--flow_type", type=str, default='tvl1', choices=['tvl1', 'warp_tvl1'])
parser.add_argument("--df_path", type=str, default='./dense_flow/', help='path to the dense_flow toolbox')
parser.add_argument("--out_format", type=str, default='zip', choices=['dir','zip'],
help='path to the dense_flow toolbox')
parser.add_argument("--ext", type=str, default='avi', choices=['avi','mp4'], help='video file extensions')
parser.add_argument("--new_width", type=int, default=320, help='resize image width')
parser.add_argument("--new_height", type=int, default=240, help='resize image height')
parser.add_argument("--num_gpu", type=int, default=2, help='number of GPU')
parser.add_argument("--resume", type=str, default='yes', choices=['yes','no'], help='resume optical flow extraction instead of overwriting')
我们的设置,全部采用默认
cmd = os.path.join(df_path + 'build/extract_warp_gpu')+' -f={} -x={} -y={} -b=20 -t=1 -d={} -s=1 -o={}'.format(
vid_path, flow_x_path, flow_y_path, dev_id, out_format)
你好,我也遇见了光流比RGB多一帧的情况,会影响到后续的工作,请问你解决这个问题了吗
您好,想请教一些问题: