paddleDetection 如何添自定义模块
收藏
怎么样才能添加并使用自定义的网络模块,在paddleDetection文档中并没有详细说明如何添加配置及使用,有大佬知道嘛?
0
收藏
请登录后评论
参考已有的config文件?参考已有的网络模块?
请问你现在找到如何构建自定义模型的方法了吗
参考已有的模型,里面有一个@***作为管理,在_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):
........写你的网络配置代码
这个自己多摸索就会了,不要放弃