AnyNet icon indicating copy to clipboard operation
AnyNet copied to clipboard

Pretrained model of AnyNet

Open chenyilun95 opened this issue 5 years ago • 12 comments

Thanks for releasing the code!

Could you please provide the pre-trained model of AnyNet in SceneFlow dataset ? Thank you a lot !

chenyilun95 avatar May 08 '19 11:05 chenyilun95

@mileyan Hi, will it be released?

yxchng avatar May 09 '19 06:05 yxchng

Watching!

abhishekmaha23 avatar Sep 02 '19 14:09 abhishekmaha23

Has released the pretrained model. Please check it. Thanks.

mileyan avatar Dec 07 '19 18:12 mileyan

@mileyan It looks like the pre-trained model is trained with spn, so one would have to compile the spn module which with pytorch 1.0+ is not possible (without some re-write to C++ extensions).

Any plans to release a pre-trained model without spn? Or do you know anyone who has gotten spn to work with pytorch 1.0+?

ghost avatar Dec 15 '19 05:12 ghost

Hi,

@mileyan It looks like the pre-trained model is trained with spn, so one would have to compile the spn module which with pytorch 1.0+ is not possible (without some re-write to C++ extensions).

Any plans to release a pre-trained model without spn? Or do you know anyone who has gotten spn to work with pytorch 1.0+?

Now our code supports pytorch 1.0. Please check it.

mileyan avatar Dec 28 '19 04:12 mileyan

Hi, Thanks for sharing the pre-trained checkpoint on SceneFlow, can you specify the metric you got? Such as EPE, 3PX and so on. I want to make sure my reimplementation approaching your result. Thanks very much.

youmi-zym avatar Feb 26 '20 01:02 youmi-zym

Hi, we evaluate the 3-pixel error on KITTI dataset.

mileyan avatar Feb 26 '20 03:02 mileyan

Hi @mileyan, How about metric on SceneFlow?

youmi-zym avatar Feb 26 '20 05:02 youmi-zym

Hi, we evaluate 3-pixel error on KITTI set. We randomly split the kitti training set 4 times and the ratio of training vs validation is 4:1. The result in the paper is the mean of results. In SceneFlow, yes, we do end point error.

mileyan avatar Feb 26 '20 21:02 mileyan

Hi @mileyan, thanks for detailing. As for SceneFlow, I have trained for 10 epoch and got EPE=3.377. Is that approaching your result? Btw, I have also extended the training schedule for 20 epoch with a constant learning rate and got EPE=3.214. Can you give the specific EPE you got, thanks a lot.

youmi-zym avatar Feb 27 '20 00:02 youmi-zym

Yes, the result is close to 3.377. It is very interesting to see that training more epochs can get better performance.

mileyan avatar Feb 28 '20 04:02 mileyan

Well, thanks a lot.

youmi-zym avatar Feb 28 '20 05:02 youmi-zym