search

lochenchou / DORN

Deep Ordinal Regression Network for Monocular Depth Estimation
11 stars 2 forks source link

Pretrained weight not available #2

Closed Youyi7 closed 2 years ago

Youyi7 commented 2 years ago

Hello, your implemention for Dorn on pytorch is very useful. While I can't train it, could you please provide the weight of your resnet?

The location of your code is in: DORN/model/resnet.py line:123 if pretrained: saved_state_dict = torch.load('/datasets/KITTI/depth_prediction/pretrained/resnet101-imagenet.pth', map_location="cpu")

Without this weight, the performance may not good, thanks very much.

lochenchou commented 2 years ago

Hi, You could download the pretrained weight from the link below http://sceneparsing.csail.mit.edu/model/pretrained_resnet/