Should I extact "tar" for FlowNet2_checkpoint.pth.tar ?

NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Other

3.08k stars 738 forks source link

Open XinyueZ opened 1 year ago

XinyueZ commented 1 year ago

hey

I want to use the pretrained model directly. I see that I cannot extract via "tar".

Shall I just call with pytorch:

config_dir = "./configs"
checkpoint_file = f'{config_dir}/FlowNet2_checkpoint.pth.tar'
flownet2 = torch.load(checkpoint_file)

thank you