demo_syncnet.py

#!/usr/bin/env python3
#-*- coding: utf-8 -*-

import time, pdb, argparse, subprocess, logging

from SyncNetInstance import *

logging.basicConfig(level=logging.INFO, format='%(asctime)s %(name)s %(levelname)s: %(message)s')
logger = logging.getLogger(__name__)

# ==================== LOAD PARAMS ====================


parser = argparse.ArgumentParser(description = "SyncNet")

parser.add_argument('--initial_model', type=str, default="data/syncnet_v2.model", help='')
parser.add_argument('--batch_size', type=int, default='20', help='')
parser.add_argument('--vshift', type=int, default='15', help='')
parser.add_argument('--videofile', type=str, default="data/example.avi", help='')
parser.add_argument('--tmp_dir', type=str, default="data/work/pytmp", help='')
parser.add_argument('--reference', type=str, default="demo", help='')

opt = parser.parse_args()


# ==================== RUN EVALUATION ====================

s = SyncNetInstance()

s.loadParameters(opt.initial_model)
logger.info("Model %s loaded.", opt.initial_model)

s.evaluate(opt, videofile=opt.videofile)