[DOC] Add custom dataset guide#306
Merged
Merged
Conversation
PeterH0323
changed the title
PeterH0323
force-pushed
the
hin/custom_dataset
branch
from
ff055d4 to
59f9338
Compare
PeterH0323
changed the title
RangeKing
reviewed
Nov 23, 2022
PeterH0323
force-pushed
the
hin/custom_dataset
branch
from
b9e1acc to
f860e0d
Compare
hhaAndroid
reviewed
Nov 26, 2022
|
|
||
| max_epochs = 200 # 训练的最大 epoch | ||
| data_root = '/path/to/data_root' # 数据集目录的绝对路径 | ||
| work_dir = './work_dirs/yolov5_s-v61_syncbn_fast_1xb8-200e_cat' # 结果保存的路径 |
Collaborator
Author
There was a problem hiding this comment.
用户会对同一个 config 进行调试，这里我暴露出来是为了让用户知道这个可以自己指定
| 因为是我们自定义的数据集，所以我们需要自己重写 config 中的部分信息，我们在 configs 目录下新建一个子集的 config 文件，这个 config 继承的是 `yolov5_s-v61_syncbn_8xb16-300e_coco.py`，数据集中的类是猫，`bs=8`，`200epoch`，可以命令为 `yolov5_s-v61_syncbn_fast_1xb8-200e_cat.py`，添加以下内容： | ||
|
|
||
| ```python | ||
| _base_ = './yolov5/yolov5_s-v61_syncbn_8xb16-300e_coco.py' |
Member
There was a problem hiding this comment.
这个配置优有点长，建议能继承的就不要重写了，否则内容太长。 我还有一个提议：咱们开一个 project/misc，这个文件夹后面专门放一些杂项，例如本文的配置就非常合适。后面一些乱七八糟的配置脚本都可以放这个文件夹下。一些正样本可视化分析代码也可以放里面。确保 readme 写清楚就行
Collaborator
Author
There was a problem hiding this comment.
这个 config 内容已经是最精简了 😢
RangeKing
reviewed
Nov 26, 2022
RangeKing
reviewed
Nov 26, 2022
RangeKing
reviewed
Nov 26, 2022
hhaAndroid
approved these changes
Nov 29, 2022
hhaAndroid
reviewed
Nov 30, 2022
RangeKing
approved these changes
Nov 30, 2022
RangeKing
left a comment
RangeKing
left a comment
Collaborator
There was a problem hiding this comment.
最后一个建议，英文文档下新建一个 custom_dataset.md，里面写上 Coming soon. Please refer to [chinese documentation](../../zh_cn/user_guides/custom_dataset.md)。并修改对应 index.est。
hhaAndroid
reviewed
Nov 30, 2022
hhaAndroid
approved these changes
Nov 30, 2022
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Motivation
Add custom dataset guide
Modification
docs/zh_cn/user_guides/custom_dataset.md