AnyNet icon indicating copy to clipboard operation
AnyNet copied to clipboard

Finetune on kitti 2015

Open Archerwz opened this issue 5 years ago • 3 comments

Hi, I wonder how you finetune on Kitti 2015 without overfitting. Because, when I tried to finetune on Kitti 2015 with random crop, the validation loss start to increase at epoch 20, further training can only make the metrics on the validation set worse. For finetune I split the 200 images to 160 as train set and 40 as val set.

Archerwz avatar Feb 22 '20 15:02 Archerwz

Hi, you can try to use Adam optimizer and use a smaller learning rate. Could you provide some training code, so I can help you do an analysis?

mileyan avatar Feb 26 '20 21:02 mileyan

So I change your network a little bit, instead of one disparity map, I form two cost volume and concat them together for later output disp map for both right and left image. I pretrain on the kitti raw dataset with official annotated depth map (contain groundtruth for both left and right) for 10 epochs which I convert them to disparity with focal length and baseline. I later finetune on kitti 2015, since it only has ground truth for left images, I warp the predicted right disparity to left and compute its lose with respect to left groundtruth.

Archerwz avatar Feb 27 '20 04:02 Archerwz

Hi, how you warp the predicted right disparity to left? I suggest you first project the left image ground truth to 3d coordinate and then use the point cloud to generate the right image ground truth.

mileyan avatar Feb 28 '20 04:02 mileyan