夯实基础系列:文本检测算法FCENet 核心代码

引言

基本原理

核心代码

# -*- encoding: utf-8 -*-
# @Author: SWHL
# @Contact: [email protected]
import torch

from torch import nn


class FCENet(nn.Moudule):
    def __init__(self,):

        self.backbone = nn.Sequential()

        self.neck = nn.Sequential()

        self.head = nn.Sequential()

    def forward(self, x):
        x = self.backbone(x)
        x = self.neck(x)
        x = self.head(x)
        return x


class FCEPostProcess():
    def __init__(self) -> None:
        pass

    def __call__(self, preds):
        pass

猜你喜欢

转载自blog.csdn.net/shiwanghualuo/article/details/129267756
今日推荐