poet icon indicating copy to clipboard operation
poet copied to clipboard

Train using custom dataset

Open z7r7y7 opened this issue 5 months ago • 9 comments

Thank you for providing this excellent project! I would like to train on my custom dataset. In the backbone.py file, I noticed that setting self[0].train_backbone to True results in a NotImplementedError. Does this mean that the backbone is currently not trainable? If I want to use Mask R-CNN as the backbone for training on my custom dataset, what steps should I follow?

z7r7y7 avatar Jan 15 '24 13:01 z7r7y7