flag=0
for k, v in enumerate(self.base):
#for k, v in enumerate(base):
#print(k,v)
x = v(x) #hxh 经过vgg或者resnet的处理?
if (k-1,k)[self.model=='resnet'] in extract[self.model]['b']:
if flag==0:
sources.append(self.L2Norm(x))
flag+=1
else:
sources.append(x)
在家里没有惯用的Linux,挨个来试一下输出
for k, v in enumerate(base):
print(k,v)
结果如下
0 vgg
1 resnet
if (k-1,k)[self.model=='resnet'] in extract[self.model]['b']:
这句是个什么意思