Customising/ modifying to training script

[manna-akash]

Hii, Its been great and appreciated to make it a public repository. I am very much interested into your work. Although, You have provided a pretrained model and working fine on unknown objects. Still its a kind of black box, if we can't trained it on our custom dataset.

Can you please help me to modify this repo to make it compatible for training, Unless, its very hard for me to cite your work.

Thanks in advance.