Training on COCO-Person dataset
See original GitHub issueHi, Thanks for the advance work you have done! This model helped me a lot. I would also like to retrain the model for one class(coco-person), and I have browsed many related issues, but I encountered some doubts.
-
Why the score is getting lower and lower?Looks like it finally converged to around 0.08.
【0】 10 || B: 1.702 | C: 3.531 | M: 2.854 | S: 0.607 | T: 8.694 || ETA: 7:15:17 || timer: 0.633 【0】100 || B: 1.460 | C: 2.346 | M: 2.624 | S: 0.353 | T: 6.784 || ETA: 7:07:40 || timer: 0.606 【0】300 || B: 1.438 | C: 1.732 | M: 2.513 | S: 0.169 | T: 5.852 || ETA: 7:07:18 || timer: 0.638 【0】800 || B: 1.522 | C: 1.586 | M: 2.468 | S: 0.106 | T: 5.682 || ETA: 7:02:46 || timer: 0.679 【0】2550 || B: 1.486 | C: 1.518 | M: 2.430 | S: 0.087 | T: 5.520 || ETA: 6:45:35 || timer: 0.672 -
How to set lr in this case? I just need to retrain on Person class.
Looking forward to your reply,Thanks a lot!
My config is:
yolact_coco_person_config = yolact_base_config.copy({
'name': 'yolact_coco_person',
'dataset': coco2017_dataset_person,
'num_classes': len(coco2017_dataset_person.class_names) + 1,
# Training params
'max_iter': 40000,
'lr': 1e-4,
'momentum': 0.9,
'decay': 5e-4,
'gamma': 0.1,
'lr_steps': (.35 * 40000, .75 * 40000, .88 * 40000, .93 * 40000),
})
I have removed other annotations and categories in the COCO dataset.
This is my dataset:
coco2017_dataset_person = dataset_base.copy({
'name': 'COCO 2017 Person',
'train_images': '../datasets/COCO2017/train2017',
'train_info': '../datasets/COCO2017/annotations/instances_train2017_person.json',
'valid_images': '../datasets/COCO2017/val2017',
'valid_info': '../datasets/COCO2017/annotations/instances_val2017_person.json',
'has_gt': True,
'class_names': ('person',),
'label_map': None,
})
Train command is:
python train.py --config=yolact_coco_person_config --resume=weights/yolact_base_54_800000.pth --start_iter=0
Issue Analytics
- State:
- Created 4 years ago
- Comments:13 (1 by maintainers)
Top GitHub Comments
@Maxinho96 i have used pedestrian dataset since my applications required pedestrains scenes
@QCHighing May I request for pre-trained model only for COCO-person class?