camenduru
/

Wav2Lip

Model card Files Files and versions Community

Wav2Lip / evaluation /scores_LSE /calculate_scores_real_videos.py

camenduru

content

170cd5b over 1 year ago

raw history blame contribute delete

No virus

1.62 kB

	#!/usr/bin/python
	#-- coding: utf-8 --

	import time, pdb, argparse, subprocess, pickle, os, gzip, glob

	from SyncNetInstance_calc_scores import *

	# ==================== PARSE ARGUMENT ====================

	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('--data_dir', type=str, default='data/work', help='');
	parser.add_argument('--videofile', type=str, default='', help='');
	parser.add_argument('--reference', type=str, default='', help='');
	opt = parser.parse_args();

	setattr(opt,'avi_dir',os.path.join(opt.data_dir,'pyavi'))
	setattr(opt,'tmp_dir',os.path.join(opt.data_dir,'pytmp'))
	setattr(opt,'work_dir',os.path.join(opt.data_dir,'pywork'))
	setattr(opt,'crop_dir',os.path.join(opt.data_dir,'pycrop'))


	# ==================== LOAD MODEL AND FILE LIST ====================

	s = SyncNetInstance();

	s.loadParameters(opt.initial_model);
	#print("Model %s loaded."%opt.initial_model);

	flist = glob.glob(os.path.join(opt.crop_dir,opt.reference,'0*.avi'))
	flist.sort()

	# ==================== GET OFFSETS ====================

	dists = []
	for idx, fname in enumerate(flist):
	offset, conf, dist = s.evaluate(opt,videofile=fname)
	print (str(dist)+" "+str(conf))

	# ==================== PRINT RESULTS TO FILE ====================

	#with open(os.path.join(opt.work_dir,opt.reference,'activesd.pckl'), 'wb') as fil:
	# pickle.dump(dists, fil)