- 输入图像img1
- 输入图像img2
- 模型选择:teacher,student
- 判断结果
- 置信度 |sigmoid(pridiction-threshhode)-0.5|*2
- 推理时间
- 得到输入图像
- 将输入图像进行预处理
- 加载入网络得到embeddings,并记录时间
- 比对,通过比较模型的treshhold判断是否为同一个人,并得到置信度
- 修改置信度公式
- 教室模型是在使用了
model = nn.DataParalle()情况下训练的,加载到CUDA应包含上面的代码。若加载到非CUDA设备,应使用这个进行处理。