Toybrick

rknn 能转换宽高都不固定的模型吗

wen0554

新手上路

积分
26
楼主
发表于 2020-9-25 15:02:14    查看: 5805|回复: 2 | [复制链接]    打印 | 只看该作者
如题,我用keras训练了一个inputs是Shape(None,None,3)的模型,我该怎么转成rknn模型。
回复

使用道具 举报

jefferyzhang

版主

积分
12928
沙发
发表于 2020-9-25 15:20:14 | 只看该作者
不能,请把宽高固定后再转rknn。
回复

使用道具 举报

wen0554

新手上路

积分
26
板凳
 楼主| 发表于 2020-9-26 10:00:20 | 只看该作者
可是我的模型就是这样的啊,如果确信大小,效果就差的远了
def res_block(x,sz,filter_sz=3,in_conv_size=1):
        xi  = x
        for i in range(in_conv_size):
                xi  = Conv2D(sz, filter_sz, activation='linear', padding='same')(xi)
                xi  = BatchNormalization()(xi)
                xi         = Activation('relu')(xi)
        xi  = Conv2D(sz, filter_sz, activation='linear', padding='same')(xi)
        xi  = BatchNormalization()(xi)
        xi         = Add()([xi,x])
        xi         = Activation('relu')(xi)
        return xi

def conv_batch(_input,fsz,csz,activation='relu',padding='same',strides=(1,1)):
        output = Conv2D(fsz, csz, activation='linear', padding=padding, strides=strides)(_input)
        output = BatchNormalization()(output)
        output = Activation(activation)(output)
        return output

def end_block(x):
        xprobs    = Conv2D(2, 3, activation='softmax', padding='same')(x)
        xbbox     = Conv2D(6, 3, activation='linear' , padding='same')(x)
        return Concatenate(3)([xprobs,xbbox])


def create_model_eccv():
       
        input_layer = Input(shape=(None,None,3),name='input')

        x = conv_batch(input_layer, 16, 3)
        x = conv_batch(x, 16, 3)
        x = MaxPooling2D(pool_size=(2,2))(x)
        x = conv_batch(x, 32, 3)
        x = res_block(x, 32)
        x = MaxPooling2D(pool_size=(2,2))(x)
        x = conv_batch(x, 64, 3)
        x = res_block(x,64)
        x = res_block(x,64)
        x = MaxPooling2D(pool_size=(2,2))(x)
        x = conv_batch(x, 64, 3)
        x = res_block(x,64)
        x = res_block(x,64)
        x = MaxPooling2D(pool_size=(2,2))(x)
        x = conv_batch(x, 128, 3)
        x = res_block(x,128)
        x = res_block(x,128)
        x = res_block(x,128)
        x = res_block(x,128)

        x = end_block(x)

        return Model(inputs=input_layer,outputs=x)
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表