2020
10-07
10-07
keras slice layer 层实现方式
注意的地方:keras中每层的输入输出的tensor是张量,比如Tensorshape是(N,H,W,C),对于tf后台,channels_lastDefineaslicelayerusingLamdalayerdefslice(x,h1,h2,w1,w2):"""Defineatensorslicefunction"""returnx[:,h1:h2,w1:w2,:]定义完slicefunction之后,利用lambdalayer添加到定义的网络中去#Addslicelayerslice_1=Lambda(slice,arguments={'h1':0,'h2':6,'w1':0,'w2':6})(slic...
继续阅读 >