-
Notifications
You must be signed in to change notification settings - Fork 513
How to train on custom Dataset? #150
-
Before someone sends me to the model training repo, please let me just explain.
I want to fine tune the existing model, say PubLayNet/faster_rcnn_R_50_FPN_3x model for my own task BUT for a Single Class, ex: text Detection only where the mapping is as {0: "Text", 1: "Title", 2: "List", 3:"Table", 4:"Figure"}
Or maybe on HJDataset where classes are {1:"Page Frame", 2:"Row", 3:"Title Region", 4:"Text Region", 5:"Title", 6:"Subtitle", 7:"Other"}.
I found this Kaggle Notebook on fine tuning with Detectron2 for fine tuning but the problem is what I have described earlier that I just want to train on 1 class.
What would be the changes that I'll have to do? How would the things_classes look like?
thing_classes= ['text'] # cfg.MODEL.ROI_HEADS.NUM_CLASSES = 1
thing_classes= ["None",'text'] # cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
thing_classes= ['text', 'None', 'None', 'None', 'None'] # cfg.MODEL.ROI_HEADS.NUM_CLASSES = 5
thing_classes= ['None', 'text', 'None', 'None', 'None', 'None'] # cfg.MODEL.ROI_HEADS.NUM_CLASSES = 6
thing_classes= ['text', 'None', 'None', 'None', 'None', 'None'] # cfg.MODEL.ROI_HEADS.NUM_CLASSES = 6
- Would it be any different if I use Layout Parser Model Config for faster_rcnn_R_50_FPN_3x instead of the default one from
Detectron2/configs/COCO-Detection/faster_rcnn_R_50_FPN_3x.yaml
Thanks in advance.
Beta Was this translation helpful? Give feedback.
All reactions
-
👀 1