diff --git a/demo.py b/demo.py index 64b659c..869f329 100644 --- a/demo.py +++ b/demo.py @@ -1,4 +1,5 @@ # -- coding: utf-8 --` +import datetime import argparse import os # engine @@ -69,6 +70,6 @@ def main(args): # inpainting parser.add_argument("--mask", type=str, default=None, help="mask of the region to inpaint on the initial image") # output name - parser.add_argument("--output", type=str, default="output.png", help="output image name") + parser.add_argument("--output", type=str, default="out_" + datetime.datetime.today().isoformat('_', 'seconds') + ".png", help="output image name") args = parser.parse_args() - main(args) + main(args) \ No newline at end of file