首页 PaddleCV 帖子详情
请教PaddleClas训练时config里的参数配置 已解决
收藏
快速回复
PaddleCV 问答图像分类 537 1
请教PaddleClas训练时config里的参数配置 已解决
收藏
快速回复
PaddleCV 问答图像分类 537 1

在【30分钟玩转PaddleClas】项目中,对flowers102(图片尺寸为image_shape: [3, 224, 224])数据集进行训练的参数配置如下

VALID:
    batch_size: 20
    num_workers: 4
    file_list: "./dataset/flowers102/val_list.txt"
    data_dir: "./dataset/flowers102/"
    shuffle_seed: 0
    transforms:
        - DecodeImage:
            to_rgb: True
            to_np: False
            channel_first: False
        - ResizeImage:
            resize_short: 256
        - CropImage:
            size: 224
        - NormalizeImage:
            scale: 1.0/255.0
            mean: [0.485, 0.456, 0.406]
            std: [0.229, 0.224, 0.225]
            order: ''
        - ToCHWImage:

 

其中- ResizeImage:
            resize_short: 256

的参数不是很理解该怎么设置。。

 

请问,现在我想训练尺寸为768x768的图片时,该怎么修改如上的参数呢?

DeepGeGe
已解决
2# 回复于2022-04
在那个PaddleClas/ppcls/data/preprocess/ops/operators.py 里面可以找到这个算子的代码: [代码]
展开
0
收藏
回复
全部评论(1)
时间顺序
DeepGeGe
#2 回复于2022-04

在那个PaddleClas/ppcls/data/preprocess/ops/operators.py 里面可以找到这个算子的代码:

class ResizeImage(object):
    """ resize image """

    def __init__(self,
                 size=None,
                 resize_short=None,
                 interpolation=None,
                 backend="cv2"):
        if resize_short is not None and resize_short > 0:
            self.resize_short = resize_short
            self.w = None
            self.h = None
        elif size is not None:
            self.resize_short = None
            self.w = size if type(size) is int else size[0]
            self.h = size if type(size) is int else size[1]
        else:
            raise OperatorParamError("invalid params for ReisizeImage for '\
                'both 'size' and 'resize_short' are None")

        self._resize_func = UnifiedResize(
            interpolation=interpolation, backend=backend)

    def __call__(self, img):
        img_h, img_w = img.shape[:2]
        if self.resize_short is not None:
            percent = float(self.resize_short) / min(img_w, img_h)
            w = int(round(img_w * percent))
            h = int(round(img_h * percent))
        else:
            w = self.w
            h = self.h
        return self._resize_func(img, (w, h))
0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户