packnet-sfm icon indicating copy to clipboard operation
packnet-sfm copied to clipboard

Loading PackNetSAN, Supervised (densified GT) pretrained model fails

Open wbkit opened this issue 3 years ago • 1 comments

Hello, I am currently trying to load the pre-trained PackNetSAN model from the checkpoint file you provide. However, the process fails due to the load.load_class function failing to find the class "PackNetSlimEnc01". Could it be that this class definition is currently missing in the repo or was this checkpoint perhaps created with a previous version of the code that used a different class structure? I'd be glad if you could provide some instruction on how to get past this issue.

Thanks in advance.

wbkit avatar Dec 02 '21 20:12 wbkit

Hi,

You can rename the depth net name from "PackNetSlimEnc01" to "PackNetSAN01"

# Parse arguments
config, state_dict = parse_test_file(ckpt_file, cfg_file)
if config.model.depth_net.name == 'PackNetSlimEnc01':
    config.model.depth_net.name = 'PackNetSAN01'

Refer to #163

jiaqixuac avatar Dec 03 '21 05:12 jiaqixuac