Spaces:
Runtime error
Runtime error
Omar Sanseviero
commited on
Commit
·
fbdd41c
1
Parent(s):
e61a53a
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
import os
|
| 2 |
os.system("git clone https://github.com/thohemp/6DRepNet")
|
| 3 |
import sys
|
| 4 |
-
sys.path.append("
|
| 5 |
|
| 6 |
from model import SixDRepNet
|
| 7 |
import math
|
|
@@ -79,7 +79,7 @@ def predict(img):
|
|
| 79 |
r_pred_deg = euler[:, 2].cpu()
|
| 80 |
utils.plot_pose_cube(frame, y_pred_deg, p_pred_deg, r_pred_deg, x_min + int(.5*(x_max-x_min)), y_min + int(.5*(y_max-y_min)), size = bbox_width)
|
| 81 |
|
| 82 |
-
|
| 83 |
|
| 84 |
|
| 85 |
iface = gr.Interface(
|
|
|
|
| 1 |
import os
|
| 2 |
os.system("git clone https://github.com/thohemp/6DRepNet")
|
| 3 |
import sys
|
| 4 |
+
sys.path.append("6DRepNet")
|
| 5 |
|
| 6 |
from model import SixDRepNet
|
| 7 |
import math
|
|
|
|
| 79 |
r_pred_deg = euler[:, 2].cpu()
|
| 80 |
utils.plot_pose_cube(frame, y_pred_deg, p_pred_deg, r_pred_deg, x_min + int(.5*(x_max-x_min)), y_min + int(.5*(y_max-y_min)), size = bbox_width)
|
| 81 |
|
| 82 |
+
return img
|
| 83 |
|
| 84 |
|
| 85 |
iface = gr.Interface(
|