question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

Training on COCO-Person dataset

See original GitHub issue

Hi, 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.

  1. 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

  2. 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:open
  • Created 4 years ago
  • Comments:13 (1 by maintainers)

github_iconTop GitHub Comments

1reaction
abhigoku10commented, Mar 22, 2020

@Maxinho96 i have used pedestrian dataset since my applications required pedestrains scenes

0reactions
sam1006commented, Oct 24, 2022

@QCHighing May I request for pre-trained model only for COCO-person class?

Read more comments on GitHub >

github_iconTop Results From Across the Web

An Introduction to the COCO Dataset - Roboflow Blog
Once the model is trained on the COCO dataset, it can be fine-tuned to learn other tasks, with a custom dataset.
Read more >
Datasets — OpenPifPaf Guide
In general, training datasets are large and require a computer with a good GPU to train and evaluate in reasonable times. Additional datasets...
Read more >
COCO dataset
COCO is a large-scale object detection, segmentation, and captioning dataset. COCO has several features: Object segmentation. Recognition in context.
Read more >
arXiv:2102.13086v2 [cs.CV] 25 Apr 2022
cabulary of concepts from all training datasets. ... example, it predicts a COCO-person separately from an. Objects365-Person, etc.
Read more >
4. Dive deep into Training a Simple Pose Model on COCO ...
The COCO keypoints dataset contains 17 keypoints for a person. Each keypoint is annotated with three numbers (x, y, v) , where x...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found