Skip to content
This repository was archived by the owner on Dec 1, 2021. It is now read-only.

Support multiple labels in networks #617

Open
yd8534976 opened this issue Nov 15, 2019 · 2 comments
Open

Support multiple labels in networks #617

yd8534976 opened this issue Nov 15, 2019 · 2 comments
Assignees

Comments

@yd8534976
Copy link
Contributor

Currently, we can only use (images, labels) pair to feed a model. It is not flexible for a complicated model.
Sometimes, we may face cases like (images, labels_1, labels_2..). We have to take it into account.

@yd8534976 yd8534976 changed the title Support multiple labels in networks [WIP] Support multiple labels in networks Nov 18, 2019
@yd8534976 yd8534976 changed the title [WIP] Support multiple labels in networks Support multiple labels in networks Nov 20, 2019
@yd8534976 yd8534976 self-assigned this Dec 4, 2019
@yd8534976
Copy link
Contributor Author

To support this feature, we have to modify blueoil/networks/base.py, blueoil/cmd/train.py, blueoil/datasets/dataset_iterator.py. But if we modify theses files directly, we would also have to change hundreds of files like #618 . If so, the PR will be too large to be reviewed. So we could add new files like blueoil/networks/base_multi.py, blueoil/cmd/train_multi.py, blueoil/datasets/dataset_iterator_multi.py instead of modifying them directly. If so, lines of code can be controlled and the PRs will be reviewable.

@yd8534976
Copy link
Contributor Author

  • create blueoil/datasets/dataset_iterator_multi.py
  • create an example dataset class, it will be blueoil/datasets/widerface_multi.py
  • create blueoil/networks/base_multi.py
  • create blueoil/networks/object_detection/yolo_utils.py
  • create blueoil/networks/object_detection/yolo_v2_multi.py
  • create blueoil/cmd/train_multi.py
  • create blueoil/configs/core/object_detection/yolo_v2_multi_widerface.py

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant