Spaces:
Running
on
T4
Running
on
T4
update app
Browse files
app.py
CHANGED
|
@@ -19,6 +19,7 @@ from perspective2d.utils.predictor import VisualizationDemo
|
|
| 19 |
import perspective2d.modeling # noqa
|
| 20 |
from perspective2d.config import get_perspective2d_cfg_defaults
|
| 21 |
from perspective2d.utils import draw_from_r_p_f_cx_cy
|
|
|
|
| 22 |
|
| 23 |
|
| 24 |
|
|
@@ -148,6 +149,12 @@ def inference(img, model_type):
|
|
| 148 |
'rad',
|
| 149 |
up_color=(0,1,0),
|
| 150 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 151 |
return Image.fromarray(pred_vis), param
|
| 152 |
|
| 153 |
examples = []
|
|
|
|
| 19 |
import perspective2d.modeling # noqa
|
| 20 |
from perspective2d.config import get_perspective2d_cfg_defaults
|
| 21 |
from perspective2d.utils import draw_from_r_p_f_cx_cy
|
| 22 |
+
from datetime import datetime
|
| 23 |
|
| 24 |
|
| 25 |
|
|
|
|
| 149 |
'rad',
|
| 150 |
up_color=(0,1,0),
|
| 151 |
)
|
| 152 |
+
print(f"""time {datetime.now().strftime("%H:%M:%S")}
|
| 153 |
+
img.shape {img.shape}
|
| 154 |
+
model_type {model_type}
|
| 155 |
+
param {param}
|
| 156 |
+
"""
|
| 157 |
+
)
|
| 158 |
return Image.fromarray(pred_vis), param
|
| 159 |
|
| 160 |
examples = []
|