add finetune single_class_dataset #351
Merged
Conversation
This file contains 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
CaitinZhao
reviewed
Sep 11, 2024
| @@ -5,18 +5,17 @@ | |||
| start_num = 1 | |||
|
|
|||
| # 获取当前目录下所有的.txt文件 | |||
| files = [f for f in os.listdir('images/train') if f.endswith(extension)] | |||
| files = [f for f in os.listdir('images/val') if f.endswith(extension)] | |||
There was a problem hiding this comment.
只需要对验证集做改名和生成json文件，这里怕误导用户
zhanghuiyao
reviewed
Sep 11, 2024
| @@ -82,4 +82,3 @@ def init_coco_format(): | |||
| # 为每个分区保存JSON文件 | |||
| with open(f"./annotations/instances_val2017.json", "w") as json_file: | |||
| json.dump(coco_format, json_file, indent=4) | |||
|
|
|||
| @@ -43,7 +42,7 @@ def xml2txt(xml, labels, name_list, img_path): | |||
| fil_name = root.find('filename').text[:-4] | |||
| if not os.path.exists(labels): # 如果路径不存在则创建 | |||
| os.mkdir(labels) | |||
| out = open(labels + './' + fil_name + '.txt', 'w+') | |||
| out = open(labels + '/' + fil_name + '.txt', 'w+') | |||
There was a problem hiding this comment.
可以换成 os.path.join(path1, path2)的方式
zhanghuiyao
previously approved these changes
Sep 11, 2024
CaitinZhao
reviewed
Sep 11, 2024
|
|
||
| for i, file in enumerate(files, start=start_num): | ||
| # 构造新的文件名，使用zfill填充前导零 | ||
| new_filename = f"{i:06d}{extension}" | ||
|
|
||
| # 使用os.rename重命名文件 | ||
| # 需根据自己的路径修改 | ||
| os.chdir('D:/Desktop/mindspore/yolo_car_detection/bddua/bdd100k/images/train') | ||
| os.chdir('D:/Desktop/mindspore/yolo_car_detection/bddua/bdd100k/images/val') |
There was a problem hiding this comment.
已修改为相对路径，并在readme中说明应在数据集的根目录下使用
yuedongli1
force-pushed
the
master
branch
2 times, most recently
from
e422565
to
d0bd2ff
Compare
CaitinZhao
approved these changes
Sep 13, 2024
zhanghuiyao
approved these changes
Sep 13, 2024
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
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.
Thank you for your contribution to the MindYOLO repo.
Before submitting this PR, please make sure:
Motivation
(Write your motivation for proposed changes here.)
Test Plan
(How should this PR be tested? Do you require special setup to run the test or repro the fixed bug?)
Related Issues and PRs
(Is this PR part of a group of changes? Link the other relevant PRs and Issues here. Use https://help.github.com/en/articles/closing-issues-using-keywords for help on GitHub syntax)