由于想继续使用pytorch1.1.0,所以可能会涉及到一些pytorch相关的修改,会注明。
0.环境
ubuntu16.04
cuda9.0
python3.6
torch==1.1.0
torchvision==0.3.0
Cython
yacs
tensorboard
future
termcolor
sklearn
tqdm
opencv-python==4.1.0.25
matplotlib
scikit-image
faiss-gpu==1.6.3
tabulate
gdown
1.下载对应版本的fast-reid
https://download.csdn.net/download/qq_35975447/12797752
2.修改
2.1 模型输出
这个地方应该是代码未更新完整,导致的接口不一致。
# kd_trainer.py
# line70outputs, targets = self.model(data)
改为:
outputs = self.model(data) # add#line80-82
with torch.no_grad():outputs_t, _ = self.model_t(data)
改为:
with torch.no_grad():outputs_t = self.model_t(data) #