-
Notifications
You must be signed in to change notification settings - Fork 7
Open
Description
Thanks for your work. I meet some problems with visualizing the depth of pred and gt. Here is the location to visualize them in
AiT/ait/code/model/depth/depth.py
Lines 157 to 159 in ca2c2d1
| for pred_d, depth_gt in results: | |
| pred_crop, gt_crop = cropping_img(pred_d, depth_gt) | |
| computed_result = eval_depth(pred_crop, gt_crop) |
for pred_d, depth_gt in results:
'''visualize 'pred_d'''
pred_crop, gt_crop = cropping_img(pred_d, depth_gt)
''' After reshaping, visualize 'pred_crop, gt_crop'''
computed_result = eval_depth(pred_crop, gt_crop)
this is cmd:
CUDA_VISIBLE_DEVICES=5,6,7 python -m torch.distributed.launch --nproc_per_node=3 code/train.py configs/swinv2b_480reso_depthonly.py --cfg-options model.task_heads.depth.vae_cfg.pretrained=vqvae_depth.pt --eval ait_joint_swinv2b.pth
However, the results of pred_d,pred_crop and gt_crop are very similar. The results of them are like this picture[The picture is almost white]

Metadata
Metadata
Assignees
Labels
No labels