tensorflow / tensorflow/models
Training only on a subset of classes present in TFRecords
Open
@pkulzc is already working on this.
Since Oct 15, 2020.
models:research:odapi
type:support
- Dominant language
- Python
- Stars
- 77.7k
- Forks
- 44.8k
- PR merge metrics
- No merged PRs in 30d
Description
There appears to be no clarity or documentation on the following issue:
Suppose one wants to train a person detector using COCO dataset. The TFRecords for COCO normally contain all object classes. Though one can create a separate set of TFRecords for persons only, is there any feature in Object Detection API where one can fine-tune only for specific classes ( counting all other categories as background ) ?
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.