oguzakif commited on
Commit
3202733
1 Parent(s): 0eb012b

get_flowNN_gradient import fix, result.mp4 output change

Browse files
Files changed (2) hide show
  1. FGT_codes/tool/video_inpainting.py +1 -1
  2. app.py +1 -1
FGT_codes/tool/video_inpainting.py CHANGED
@@ -9,7 +9,7 @@ warnings.filterwarnings("ignore")
9
 
10
  import cvbase
11
  from torchvision.transforms import ToTensor
12
- from tool import get_flowNN_gradient
13
  from tool.utils.Poisson_blend_img import Poisson_blend_img
14
  from tool.utils.region_fill import regionfill
15
  from importlib import import_module
 
9
 
10
  import cvbase
11
  from torchvision.transforms import ToTensor
12
+ from tool.get_flowNN_gradient import get_flowNN_gradient
13
  from tool.utils.Poisson_blend_img import Poisson_blend_img
14
  from tool.utils.region_fill import regionfill
15
  from importlib import import_module
app.py CHANGED
@@ -67,7 +67,7 @@ parser.add_argument(
67
  parser.add_argument(
68
  '--path_mask', default='/myData/dilateAnnotations_4/walking', help="mask for object removal")
69
  parser.add_argument(
70
- '--outroot', default='quick_start/walking3', help="output directory")
71
  parser.add_argument('--consistencyThres', dest='consistencyThres', default=5, type=float,
72
  help='flow consistency error threshold')
73
  parser.add_argument('--alpha', dest='alpha', default=0.1, type=float)
 
67
  parser.add_argument(
68
  '--path_mask', default='/myData/dilateAnnotations_4/walking', help="mask for object removal")
69
  parser.add_argument(
70
+ '--outroot', default=os.path.abspath(project_name), help="output directory")
71
  parser.add_argument('--consistencyThres', dest='consistencyThres', default=5, type=float,
72
  help='flow consistency error threshold')
73
  parser.add_argument('--alpha', dest='alpha', default=0.1, type=float)