get_flowNN_gradient import fix, result.mp4 output change
Browse files- FGT_codes/tool/video_inpainting.py +1 -1
- 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=
|
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)
|