SH Zabihifar

Train on custom objects

16

Your work is very interesting. How can I train for new custom objects or test your trained model on new objects?

How can I solve this problem? Traceback (most recent call last): File "run.py", line 103, in final = verify_6D_poses(dets_6d, model_map, bench.cam, f.color) File "/home/~/ssd-6d-master/rendering/utils.py", line 223, in verify_6D_poses ren.draw_model(model, pose)...

Problem for running the Trained Model

2

Hi, when I want to run this line of code from latentfusion.recon.inference import LatentFusionModel I am getting this error: cannot import name 'autocast' from 'torch.cuda.amp' This installation also doesn't help...