提交 e55fd759 编写于 作者: M michaelowenliu

change dygraph to paddleseg api

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