首页 PaddleCV 帖子详情
paddleDetection 如何添自定义模块
收藏
快速回复
PaddleCV 问答目标检测目标识别跟踪 608 4
paddleDetection 如何添自定义模块
收藏
快速回复
PaddleCV 问答目标检测目标识别跟踪 608 4

怎么样才能添加并使用自定义的网络模块,在paddleDetection文档中并没有详细说明如何添加配置及使用,有大佬知道嘛?

0
收藏
回复
全部评论(4)
时间顺序
AIStudio810260
#2 回复于2021-05

参考已有的config文件?参考已有的网络模块?

0
回复
jiduoweichen
#3 回复于2021-11

请问你现在找到如何构建自定义模型的方法了吗

 

0
回复
山茶油0
#4 回复于2021-11

参考已有的模型,里面有一个@***作为管理,在_init_.py里添加你定义的模型。举个例,假如你要在PaddleDetection /ppdet /modeling /architectures这个里面新加入一个模型结构,比如加入yolov3,你需要 用到@register,再用@classmethod,用于配置config。

最后要在_init_.py这个文件写下from . import yolo   ;from .yolo import *

@register

class YOLOv3(BaseArch):

..........写你的结构代码

@classmethod

def from_config(cls, cfg, *args, **kwargs):

........写你的网络配置代码

 

 

1
回复
山茶油0
#5 回复于2021-11

这个自己多摸索就会了,不要放弃

 

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户