|
@@ -1,4 +1,4 @@
|
|
|
-# Copyright (c) 2020 PaddlePaddle Authors. All Rights Reserved.
|
|
|
+# Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
|
|
|
#
|
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
|
# you may not use this file except in compliance with the License.
|
|
@@ -30,23 +30,23 @@ import paddle.distributed as dist
|
|
|
from paddle.distributed import fleet
|
|
|
from paddle import amp
|
|
|
from paddle.static import InputSpec
|
|
|
-from paddlers.third_party.ppdet.optimizer import ModelEMA
|
|
|
-
|
|
|
-from paddlers.third_party.ppdet.core.workspace import create
|
|
|
-from paddlers.third_party.ppdet.modeling.architectures.meta_arch import BaseArch
|
|
|
-from paddlers.third_party.ppdet.utils.checkpoint import load_weight, load_pretrain_weight
|
|
|
-from paddlers.third_party.ppdet.utils.visualizer import visualize_results, save_result
|
|
|
-from paddlers.third_party.ppdet.metrics import Metric, COCOMetric, VOCMetric, WiderFaceMetric, get_infer_results, KeyPointTopDownCOCOEval, KeyPointTopDownMPIIEval
|
|
|
-from paddlers.third_party.ppdet.metrics import RBoxMetric, JDEDetMetric, SNIPERCOCOMetric
|
|
|
-from paddlers.third_party.ppdet.data.source.sniper_coco import SniperCOCODataSet
|
|
|
-from paddlers.third_party.ppdet.data.source.category import get_categories
|
|
|
-from paddlers.third_party.ppdet.utils import stats
|
|
|
-from paddlers.third_party.ppdet.utils import profiler
|
|
|
+from paddlers.models.ppdet.optimizer import ModelEMA
|
|
|
+
|
|
|
+from paddlers.models.ppdet.core.workspace import create
|
|
|
+from paddlers.models.ppdet.modeling.architectures.meta_arch import BaseArch
|
|
|
+from paddlers.models.ppdet.utils.checkpoint import load_weight, load_pretrain_weight
|
|
|
+from paddlers.models.ppdet.utils.visualizer import visualize_results, save_result
|
|
|
+from paddlers.models.ppdet.metrics import Metric, COCOMetric, VOCMetric, WiderFaceMetric, get_infer_results, KeyPointTopDownCOCOEval, KeyPointTopDownMPIIEval
|
|
|
+from paddlers.models.ppdet.metrics import RBoxMetric, JDEDetMetric, SNIPERCOCOMetric
|
|
|
+from paddlers.models.ppdet.data.source.sniper_coco import SniperCOCODataSet
|
|
|
+from paddlers.models.ppdet.data.source.category import get_categories
|
|
|
+from paddlers.models.ppdet.utils import stats
|
|
|
+from paddlers.models.ppdet.utils import profiler
|
|
|
|
|
|
from .callbacks import Callback, ComposeCallback, LogPrinter, Checkpointer, WiferFaceEval, VisualDLWriter, SniperProposalsGenerator
|
|
|
from .export_utils import _dump_infer_config, _prune_input_spec
|
|
|
|
|
|
-from paddlers.third_party.ppdet.utils.logger import setup_logger
|
|
|
+from paddlers.models.ppdet.utils.logger import setup_logger
|
|
|
logger = setup_logger('ppdet.engine')
|
|
|
|
|
|
__all__ = ['Trainer']
|