提交 e55fd759 编写于 作者: M michaelowenliu

change dygraph to paddleseg api

上级 30ec28bc
...@@ -17,11 +17,11 @@ import argparse ...@@ -17,11 +17,11 @@ import argparse
import paddle.fluid as fluid import paddle.fluid as fluid
from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.dygraph.parallel import ParallelEnv
from dygraph.datasets import DATASETS from paddleseg.datasets import DATASETS
import dygraph.transforms as T import paddleseg.transforms as T
from dygraph.cvlibs import manager from paddleseg.cvlibs import manager
from dygraph.utils import get_environ_info from paddleseg.utils import get_environ_info
from dygraph.core import infer from paddleseg.core import infer
def parse_args(): def parse_args():
......
...@@ -20,8 +20,8 @@ import paddle.fluid as fluid ...@@ -20,8 +20,8 @@ import paddle.fluid as fluid
import cv2 import cv2
import tqdm import tqdm
from dygraph import utils from paddleseg import utils
import dygraph.utils.logger as logger import paddleseg.utils.logger as logger
def mkdir(path): def mkdir(path):
......
...@@ -22,10 +22,10 @@ from paddle.fluid.io import DataLoader ...@@ -22,10 +22,10 @@ from paddle.fluid.io import DataLoader
from paddle.io import DistributedBatchSampler from paddle.io import DistributedBatchSampler
import paddle.nn.functional as F import paddle.nn.functional as F
import dygraph.utils.logger as logger import paddleseg.utils.logger as logger
from dygraph.utils import load_pretrained_model from paddleseg.utils import load_pretrained_model
from dygraph.utils import resume from paddleseg.utils import resume
from dygraph.utils import Timer, calculate_eta from paddleseg.utils import Timer, calculate_eta
from .val import evaluate from .val import evaluate
......
...@@ -22,9 +22,9 @@ import paddle.fluid as fluid ...@@ -22,9 +22,9 @@ import paddle.fluid as fluid
import paddle.nn.functional as F import paddle.nn.functional as F
import paddle import paddle
import dygraph.utils.logger as logger import paddleseg.utils.logger as logger
from dygraph.utils import ConfusionMatrix from paddleseg.utils import ConfusionMatrix
from dygraph.utils import Timer, calculate_eta from paddleseg.utils import Timer, calculate_eta
def evaluate(model, def evaluate(model,
......
...@@ -17,12 +17,10 @@ import argparse ...@@ -17,12 +17,10 @@ import argparse
import paddle.fluid as fluid import paddle.fluid as fluid
from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.dygraph.parallel import ParallelEnv
import dygraph import paddleseg
from dygraph.cvlibs import manager from paddleseg.cvlibs import manager
from dygraph.utils import get_environ_info from paddleseg.utils import get_environ_info, logger, Config
from dygraph.utils import logger from paddleseg.core import train
from dygraph.utils import Config
from dygraph.core import train
def parse_args(): def parse_args():
......
...@@ -17,11 +17,10 @@ import argparse ...@@ -17,11 +17,10 @@ import argparse
import paddle.fluid as fluid import paddle.fluid as fluid
from paddle.fluid.dygraph.parallel import ParallelEnv from paddle.fluid.dygraph.parallel import ParallelEnv
import dygraph import paddleseg
from dygraph.cvlibs import manager from paddleseg.cvlibs import manager
from dygraph.utils import get_environ_info from paddleseg.utils import get_environ_info, Config
from dygraph.utils import Config from paddleseg.core import evaluate
from dygraph.core import evaluate
def parse_args(): def parse_args():
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册