提交 99974207 编写于 作者: Z zhiqiu

follow comments

上级 b8897cc6
......@@ -22,6 +22,7 @@ from functools import reduce
from PIL import Image
import cv2
import numpy as np
import paddle
import paddle.fluid as fluid
from visualize import visualize_box_mask
......@@ -629,6 +630,7 @@ def print_arguments(args):
if __name__ == '__main__':
paddle.enable_static()
parser = argparse.ArgumentParser(description=__doc__)
parser.add_argument(
"--model_dir",
......
......@@ -146,5 +146,7 @@ class TestDataset(unittest.TestCase):
if __name__ == '__main__':
import paddle
paddle.enable_static()
logging.basicConfig()
unittest.main()
......@@ -168,4 +168,6 @@ class TestReader(unittest.TestCase):
if __name__ == '__main__':
import paddle
paddle.enable_static()
unittest.main()
......@@ -112,4 +112,7 @@ class TestReaderYAML(unittest.TestCase):
if __name__ == '__main__':
import paddle
paddle.enable_static()
unittest.main()
......@@ -120,4 +120,6 @@ class TestLeftPoolOp(TestRightPoolOp):
if __name__ == "__main__":
import paddle
paddle.enable_static()
unittest.main()
......@@ -91,4 +91,5 @@ class TestSSD(TestFasterRCNN):
if __name__ == '__main__':
paddle.enable_static()
unittest.main()
......@@ -26,6 +26,7 @@ import numpy as np
from collections import OrderedDict
from paddleslim.dist.single_distiller import merge, l2_loss
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
from ppdet.data.reader import create_reader
......@@ -369,6 +370,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"-r",
......
......@@ -29,6 +29,7 @@ from paddleslim.dist.single_distiller import merge, l2_loss
from paddleslim.prune import Pruner
from paddleslim.analysis import flops
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
from ppdet.data.reader import create_reader
......@@ -335,6 +336,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"-t",
......
......@@ -27,6 +27,7 @@ import numpy as np
import datetime
from collections import deque
import paddle
from paddle import fluid
from ppdet.experimental import mixed_precision_context
......@@ -377,6 +378,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"-r",
......
......@@ -23,6 +23,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 3)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
import paddle.fluid as fluid
from paddleslim.prune import Pruner
from paddleslim.analysis import flops
......@@ -198,6 +199,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--json_eval",
......
......@@ -22,6 +22,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 3)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -136,6 +137,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--output_dir",
......
......@@ -26,6 +26,7 @@ import glob
import numpy as np
from PIL import Image
import paddle
from paddle import fluid
from paddleslim.prune import Pruner
from paddleslim.analysis import flops
......@@ -223,6 +224,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--infer_dir",
......
......@@ -29,6 +29,7 @@ import datetime
from collections import deque
from paddleslim.prune import Pruner
from paddleslim.analysis import flops
import paddle
from paddle import fluid
from ppdet.experimental import mixed_precision_context
......@@ -346,6 +347,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"-r",
......
......@@ -23,6 +23,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 3)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
import paddle.fluid as fluid
from ppdet.utils.eval_utils import parse_fetches, eval_run, eval_results, json_eval_results
......@@ -155,6 +156,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--json_eval",
......
......@@ -22,6 +22,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 3)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -105,6 +106,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--output_dir",
......
......@@ -26,6 +26,7 @@ import glob
import numpy as np
from PIL import Image
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -167,6 +168,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--infer_dir",
......
......@@ -28,6 +28,7 @@ import datetime
from collections import deque
import shutil
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -285,6 +286,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"-r",
......
......@@ -27,6 +27,7 @@ import numpy as np
import datetime
from collections import deque
import paddle
from paddle import fluid
from ppdet.experimental import mixed_precision_context
from ppdet.core.workspace import load_config, merge_config, create
......@@ -166,6 +167,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--output_eval",
......
......@@ -22,6 +22,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
import paddle.fluid as fluid
from ppdet.utils.eval_utils import parse_fetches, eval_run, eval_results, json_eval_results
......@@ -164,7 +165,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
......@@ -23,6 +23,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -226,7 +227,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
......@@ -22,6 +22,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -94,7 +95,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
......@@ -23,6 +23,7 @@ parent_path = os.path.abspath(os.path.join(__file__, *(['..'] * 2)))
if parent_path not in sys.path:
sys.path.append(parent_path)
import paddle
import paddle.fluid as fluid
import numpy as np
import cv2
......@@ -270,7 +271,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
......@@ -27,6 +27,7 @@ import numpy as np
import six
from PIL import Image
import paddle
from paddle import fluid
from ppdet.core.workspace import load_config, merge_config, create
......@@ -226,6 +227,7 @@ def main():
if __name__ == '__main__':
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
"--infer_dir",
......
......@@ -30,6 +30,7 @@ import six
from collections import deque
from paddle.fluid import profiler
import paddle
from paddle import fluid
from paddle.fluid.layers.learning_rate_scheduler import _decay_step_counter
from paddle.fluid.optimizer import ExponentialMovingAverage
......@@ -313,7 +314,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
......@@ -28,6 +28,7 @@ import random
import datetime
import six
from collections import deque
import paddle
from paddle.fluid import profiler
from paddle import fluid
......@@ -307,7 +308,6 @@ def main():
if __name__ == '__main__':
import paddle
paddle.enable_static()
parser = ArgsParser()
parser.add_argument(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册