def main():
if os.path.isdir(opt.media_path):
files = util.Traversal(opt.media_path)
else:
files = [opt.media_path]
if opt.mode == 'add':
netS = loadmodel.bisenet(opt, 'roi')
for file in files:
opt.media_path = file
if util.is_img(file):
core.addmosaic_img(opt, netS)
elif util.is_video(file):
core.addmosaic_video(opt, netS)
util.clean_tempfiles(opt, tmp_init=False)
else:
print('This type of file is not supported')
util.clean_tempfiles(opt, tmp_init=False)
elif opt.mode == 'clean':
netM = loadmodel.bisenet(opt, 'mosaic')
if opt.traditional:
netG = None
elif opt.netG == 'video':
netG = loadmodel.video(opt)
else:
netG = loadmodel.pix2pix(opt)
for file in files:
opt.media_path = file
if util.is_img(file):
core.cleanmosaic_img(opt, netG, netM)
elif util.is_video(file):
if opt.netG == 'video' and not opt.traditional:
core.cleanmosaic_video_fusion(opt, netG, netM)
else:
core.cleanmosaic_video_byframe(opt, netG, netM)
util.clean_tempfiles(opt, tmp_init=False)
else:
print('This type of file is not supported')
elif opt.mode == 'style':
netG = loadmodel.style(opt)
for file in files:
opt.media_path = file
if util.is_img(file):
core.styletransfer_img(opt, netG)
elif util.is_video(file):
core.styletransfer_video(opt, netG)
util.clean_tempfiles(opt, tmp_init=False)
else:
print('This type of file is not supported')
util.clean_tempfiles(opt, tmp_init=False)
def main():
if opt.mode == 'add':
if util.is_img(opt.media_path):
core.addmosaic_img(opt)
elif util.is_video(opt.media_path):
core.addmosaic_video(opt)
else:
print('This type of file is not supported')
elif opt.mode == 'clean':
if util.is_img(opt.media_path):
core.cleanmosaic_img(opt)
elif util.is_video(opt.media_path):
if opt.netG == 'video':
core.cleanmosaic_video_fusion(opt)
else:
core.cleanmosaic_video_byframe(opt)
else:
print('This type of file is not supported')
util.clean_tempfiles(tmp_init=False)