Please note for users using training code that the weights obtained from retraining need to be passed ./mlsd/mlsd_pytorch/models/mbv2_mlsd_large.py is used to load and infer, rather than ./mlsd/models/mbv2_mlsd_large.py. The latter includes some special processing for the tf model, as well as differences in the input channel (the latter has an additional alpha channel), and the weights of the two are not compatible.
Please note for users using training code that the weights obtained from retraining need to be passed
./mlsd/mlsd_pytorch/models/mbv2_mlsd_large.pyis used to load and infer, rather than./mlsd/models/mbv2_mlsd_large.py. The latter includes some special processing for the tf model, as well as differences in the input channel (the latter has an additional alpha channel), and the weights of the two are not compatible.