SebastienLinker

You can do single image inference. Have a look at the dataloaders. Also, the inference function also computes the nme, but you can fill the ground truth with dummy values...

Yes, there is something wrong in that model / code. I tried `model.load_state_dict(state_dict)` instead of `model.module.load_state_dict(state_dict)`. It seems to be working

@wqz960 I retrained on my own dataset

I solved this error by adding the flag '--config=monolithic' when building tensorflow