Commit ·
6206b8b
1
Parent(s): d70573e
face recognition
Browse files
app/Hackathon_setup/face_recognition.py
CHANGED
|
@@ -176,8 +176,9 @@ def get_face_class(img1):
|
|
| 176 |
ckpt = torch.load(ckpt_path, map_location=device)
|
| 177 |
myModel.load_state_dict(ckpt['net_dict'])
|
| 178 |
myModel.eval()
|
|
|
|
| 179 |
|
| 180 |
-
img_tensor = transform1(img1).unsqueeze(0)
|
| 181 |
|
| 182 |
with torch.no_grad():
|
| 183 |
embedding = myModel.forward_once(img_tensor)
|
|
|
|
| 176 |
ckpt = torch.load(ckpt_path, map_location=device)
|
| 177 |
myModel.load_state_dict(ckpt['net_dict'])
|
| 178 |
myModel.eval()
|
| 179 |
+
myModel = myModel.float()
|
| 180 |
|
| 181 |
+
img_tensor = transform1(img1).unsqueeze(0).to(device).float()
|
| 182 |
|
| 183 |
with torch.no_grad():
|
| 184 |
embedding = myModel.forward_once(img_tensor)
|