推理的时候将图片转换为np.float32类型
def Forward(self, image, ir_box):
self.Preprocess(image, ir_box)
self.living_ir_img = np.float32(self.living_ir_img)
test_output = self.net.forward(torch.from_numpy(self.living_ir_img))
ir_map = np.squeeze(test_output[0])
score = torch.mean(ir_map)
return score