About making 'best_obj.pt' on my own about g2l_net HOT 3 OPEN

I have trained obj_6 by yolov3 and G2L_NET, but the test code(test_linemod.py) doesn't work.

The error is:
RuntimeError: Error(s) in loading state_dict for Darknet:
Missing key(s) in state_dict: "module_list.0.conv_0.weight", xxxxxxxxxxxxxxxxxxxxx
Unexpected key(s) in state_dict: "total_ops", "total_params",xxxxxxxxxxxxxxxxxxxx

from g2l_net.

@XLXIAOLONG @DC1991 How you trained G2L?

I tried to run trained_G2l.py but i had this problem:
RuntimeError: Given groups=1, weight of size 64 3 1 1, expected input[64, 6, 1, 1500] to have 3 channels, but got 6 channels instead

Notes that I followed all the instructions and i download the training data and put it in the right folder.

from g2l_net.

@XLXIAOLONG @DC1991 How you trained G2L?

I tried to run trained_G2l.py but i had this problem: RuntimeError: Given groups=1, weight of size 64 3 1 1, expected input[64, 6, 1, 1500] to have 3 channels, but got 6 channels instead

Notes that I followed all the instructions and i download the training data and put it in the right folder.

Sorry for the late reply. Because some reasons, I haven't train G2L on my own dataset. But i had the same problem. The reason seem to be the point cloud. The 6 channels contain color information, but G2L need coordinate only. Extract 3 channels position from point cloud may solve the problem.

Best wishes!
xiaolong

from g2l_net.