pt2safe
Browse files
app.py
CHANGED
|
@@ -524,7 +524,7 @@ def get_reconstructed_scene(outdir, filelist, schedule='linear', niter=300, min_
|
|
| 524 |
mobilesamv2.to(device=device)
|
| 525 |
mobilesamv2.eval()
|
| 526 |
|
| 527 |
-
YOLO8_CKP='./checkpoints/ObjectAwareModel.
|
| 528 |
yolov8 = ObjectAwareModel(YOLO8_CKP)
|
| 529 |
|
| 530 |
|
|
|
|
| 524 |
mobilesamv2.to(device=device)
|
| 525 |
mobilesamv2.eval()
|
| 526 |
|
| 527 |
+
YOLO8_CKP='./checkpoints/ObjectAwareModel.pt'
|
| 528 |
yolov8 = ObjectAwareModel(YOLO8_CKP)
|
| 529 |
|
| 530 |
|