Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ def depth_normal(img):
|
|
47 |
cv_image = np.array(img)
|
48 |
img = cv2.cvtColor(cv_image, cv2.COLOR_BGR2RGB)
|
49 |
intrinsic = [1000.0, 1000.0, img.shape[1]/2, img.shape[0]/2]
|
50 |
-
rgb_input, cam_models_stacks, pad, label_scale_factor = transform_test_data_scalecano(
|
51 |
|
52 |
with torch.no_grad():
|
53 |
pred_depth, pred_depth_scale, scale, output = get_prediction(
|
|
|
47 |
cv_image = np.array(img)
|
48 |
img = cv2.cvtColor(cv_image, cv2.COLOR_BGR2RGB)
|
49 |
intrinsic = [1000.0, 1000.0, img.shape[1]/2, img.shape[0]/2]
|
50 |
+
rgb_input, cam_models_stacks, pad, label_scale_factor = transform_test_data_scalecano(img, intrinsic, cfg.data_basic)
|
51 |
|
52 |
with torch.no_grad():
|
53 |
pred_depth, pred_depth_scale, scale, output = get_prediction(
|