For inference, change the pretrained weight path and the img path
The pretrained weight can be downloaded from the link: https://drive.google.com/file/d/1AfidvEDCjOB25Y9uFyA3MkNSft7uLv4T/view?usp=drive_link
Run the following command for inferencing a single image.
python forward.py
Note: the pretrained weight is based on Franka Emika Panda robot.