| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- # copyright (c) 2020 PaddlePaddle Authors. All Rights Reserve.
- #
- # Licensed under the Apache License, Version 2.0 (the "License");
- # you may not use this file except in compliance with the License.
- # You may obtain a copy of the License at
- #
- # http://www.apache.org/licenses/LICENSE-2.0
- #
- # Unless required by applicable law or agreed to in writing, software
- # distributed under the License is distributed on an "AS IS" BASIS,
- # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- # See the License for the specific language governing permissions and
- # limitations under the License.
- import os
- import os.path as osp
- from .cls_transforms import ClsTransform
- from .det_transforms import DetTransform
- from .seg_transforms import SegTransform
- def visualize(dataset, index=0, steps=3, save_dir='vdl_output'):
- '''对数据预处理/增强中间结果进行可视化。
- 可使用VisualDL查看中间结果:
- 1. VisualDL启动方式: visualdl --logdir vdl_output --port 8001
- 2. 浏览器打开 https://0.0.0.0:8001即可,
- 其中0.0.0.0为本机访问,如为远程服务, 改成相应机器IP
-
- Args:
- dataset (paddlex.datasets): 数据集读取器。
- index (int): 对数据集中的第index张图像进行可视化。默认为0
- steps (int): 数据预处理/增强的次数。默认为3。
- save_dir (str): 日志保存的路径。默认为'vdl_output'。
- '''
- transforms = dataset.transforms
- if not osp.isdir(save_dir):
- if osp.exists(save_dir):
- os.remove(save_dir)
- os.makedirs(save_dir)
- for i, data in enumerate(dataset.iterator()):
- if i == index:
- break
- from visualdl import LogWriter
- vdl_save_dir = osp.join(save_dir, 'image_transforms')
- vdl_writer = LogWriter(vdl_save_dir)
- data.append(vdl_writer)
- for s in range(steps):
- if s != 0:
- data.pop()
- data.append(s)
- transforms(*data)
-
|