research problem

[wexj]

Hello, I would like to apply the mask completion function of this network to robot grasping. I need the network to output the category of each object, but I encountered difficulties while modifying the code. I am not sure if it is possible to replace the U-net network with another network or if there is any other better solution. I hope you can provide me with ideas or help me solve this research problem. Thank you very much