From 9efb01006cadf2acecc7667923b9846a75cfa6df Mon Sep 17 00:00:00 2001 From: sunyanfang01 Date: Fri, 8 May 2020 16:45:40 +0800 Subject: [PATCH] add faster_rcnn.py and yolov3.py code --- paddlex/cv/models/faster_rcnn.py | 7 ++++--- paddlex/cv/models/yolo_v3.py | 7 ++++--- 2 files changed, 8 insertions(+), 6 deletions(-) diff --git a/paddlex/cv/models/faster_rcnn.py b/paddlex/cv/models/faster_rcnn.py index 47dbd75..b304f87 100644 --- a/paddlex/cv/models/faster_rcnn.py +++ b/paddlex/cv/models/faster_rcnn.py @@ -176,7 +176,7 @@ class FasterRCNN(BaseAPI): log_interval_steps (int): 训练日志输出间隔（单位：迭代次数）。默认为20。 save_dir (str): 模型保存路径。默认值为'output'。 pretrain_weights (str): 若指定为路径时，则加载路径下预训练模型；若为字符串'IMAGENET'， - 则自动下载在ImageNet图片数据上预训练的模型权重；若为None，则不使用预训练模型。默认为None。 + 则自动下载在ImageNet图片数据上预训练的模型权重；若为None，则不使用预训练模型。默认为'IMAGENET'。 optimizer (paddle.fluid.optimizer): 优化器。当该参数为None时，使用默认优化器： fluid.layers.piecewise_decay衰减策略，fluid.optimizer.Momentum优化方法。 learning_rate (float): 默认优化器的初始学习率。默认为0.0025。 @@ -194,11 +194,12 @@ class FasterRCNN(BaseAPI): if metric is None: if isinstance(train_dataset, paddlex.datasets.CocoDetection): metric = 'COCO' - elif isinstance(train_dataset, paddlex.datasets.VOCDetection): + elif isinstance(train_dataset, paddlex.datasets.VOCDetection) or \ + isinstance(train_dataset, paddlex.datasets.EasyDataDet): metric = 'VOC' else: raise ValueError( - "train_dataset should be datasets.VOCDetection or datasets.COCODetection." + "train_dataset should be datasets.VOCDetection or datasets.COCODetection or datasets.EasyDataDet." ) assert metric in ['COCO', 'VOC'], "Metric only support 'VOC' or 'COCO'" self.metric = metric diff --git a/paddlex/cv/models/yolo_v3.py b/paddlex/cv/models/yolo_v3.py index 9b296cb..d65224a 100644 --- a/paddlex/cv/models/yolo_v3.py +++ b/paddlex/cv/models/yolo_v3.py @@ -175,7 +175,7 @@ class YOLOv3(BaseAPI): log_interval_steps (int): 训练日志输出间隔（单位：迭代次数）。默认为10。 save_dir (str): 模型保存路径。默认值为'output'。 pretrain_weights (str): 若指定为路径时，则加载路径下预训练模型；若为字符串'IMAGENET'， - 则自动下载在ImageNet图片数据上预训练的模型权重；若为None，则不使用预训练模型。默认为None。 + 则自动下载在ImageNet图片数据上预训练的模型权重；若为None，则不使用预训练模型。默认为'IMAGENET'。 optimizer (paddle.fluid.optimizer): 优化器。当该参数为None时，使用默认优化器： fluid.layers.piecewise_decay衰减策略，fluid.optimizer.Momentum优化方法。 learning_rate (float): 默认优化器的学习率。默认为1.0/8000。 @@ -198,11 +198,12 @@ class YOLOv3(BaseAPI): if metric is None: if isinstance(train_dataset, paddlex.datasets.CocoDetection): metric = 'COCO' - elif isinstance(train_dataset, paddlex.datasets.VOCDetection): + elif isinstance(train_dataset, paddlex.datasets.VOCDetection) or \ + isinstance(train_dataset, paddlex.datasets.EasyDataDet): metric = 'VOC' else: raise ValueError( - "train_dataset should be datasets.VOCDetection or datasets.COCODetection." + "train_dataset should be datasets.VOCDetection or datasets.COCODetection or datasets.EasyDataDet." ) assert metric in ['COCO', 'VOC'], "Metric only support 'VOC' or 'COCO'" self.metric = metric -- GitLab