为什么使用pycaffe提取时特征会被覆盖？

Question

我发现当我使用 pycaffe 提取特征时，特征被覆盖了。我的代码如下：

tImg_1 = misc.imread('1.jpg')
tImg_1 = tImg_1[:,:,::-1] # color channel swap
tImg_2 = misc.imread('2.jpg')
tImg_2 = tImg_2[:,:,::-1] # color channel swap

tImg_1 = (np.float32(tImg_1)- 127.5)/128 # mean substruction
tImg_2 = (np.float32(tImg_2)- 127.5)/128 # mean substruction

tI_1 = np.moveaxis(tImg_1, 0, 1) # Transpose
tI_2 = np.moveaxis(tImg_2, 0, 1) # Transpose

# Extract features
tI_1 = np.reshape(tImg_1, (1, tImg_1.shape[2], tImg_1.shape[0], tImg_1.shape[1]))
tI_2 = np.reshape(tImg_2, (1, tImg_2.shape[2], tImg_2.shape[0],     tImg_2.shape[1]))                

net.blobs['data'].data[...] = tI_1
net.forward() 
fts_1 = net.blobs['fc5'].data
print(fts_1[0, 0])                 

net.blobs['data'].data[...] = tI_2
net.forward() 
fts_2 = net.blobs['fc5'].data                  

print(fts_2[0, 0])                 
print(fts_1[0, 0])

执行此操作会提供以下输出：

0.508398
-0.176945
-0.176945

这意味着 fts_1 的值被 fts_2 覆盖。我怎样才能避免这个问题？

Answer 1

fts_1 只是指向 net.blobs['fc5'].data。您需要对对象进行深层复制。所以你的第一个作业应该是 fts_1 = copy.deepcopy(net.blobs['fc5'].data)

为什么使用pycaffe提取时特征会被覆盖？

Why the features are overwritten when extracted using pycaffe?

caffe

pycaffe