当前位置: 代码迷 >> 综合 >> fast-reid复现知识蒸馏部分(Market1501)--- 2020.08.31版本
  详细解决方案

fast-reid复现知识蒸馏部分(Market1501)--- 2020.08.31版本

热度:113   发布时间:2023-10-26 00:16:02.0

由于想继续使用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) # 
  相关解决方案