Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
谢爱卿
yolov4-pytorch
提交
d3d01e29
Y
yolov4-pytorch
项目概览
谢爱卿
/
yolov4-pytorch
与 Fork 源项目一致
从无法访问的项目Fork
通知
3
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
Y
yolov4-pytorch
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
d3d01e29
编写于
9月 21, 2021
作者:
B
Bubbliiiing
提交者:
GitHub
9月 21, 2021
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Delete voc2yolo4.py
上级
15cce832
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
0 addition
and
61 deletion
+0
-61
VOCdevkit/VOC2007/voc2yolo4.py
VOCdevkit/VOC2007/voc2yolo4.py
+0
-61
未找到文件。
VOCdevkit/VOC2007/voc2yolo4.py
已删除
100644 → 0
浏览文件 @
15cce832
#----------------------------------------------------------------------#
# 验证集的划分在train.py代码里面进行
# test.txt和val.txt里面没有内容是正常的。训练不会使用到。
#----------------------------------------------------------------------#
'''
#--------------------------------注意----------------------------------#
如果在pycharm中运行时提示:
FileNotFoundError: [WinError 3] 系统找不到指定的路径。: './VOCdevkit/VOC2007/Annotations'
这是pycharm运行目录的问题,最简单的方法是将该文件复制到根目录后运行。
可以查询一下相对目录和根目录的概念。在VSCODE中没有这个问题。
#--------------------------------注意----------------------------------#
'''
import
os
import
random
random
.
seed
(
0
)
xmlfilepath
=
r
'./VOCdevkit/VOC2007/Annotations'
saveBasePath
=
r
"./VOCdevkit/VOC2007/ImageSets/Main/"
#----------------------------------------------------------------------#
# 想要增加测试集修改trainval_percent
# train_percent不需要修改
#----------------------------------------------------------------------#
trainval_percent
=
1
train_percent
=
1
temp_xml
=
os
.
listdir
(
xmlfilepath
)
total_xml
=
[]
for
xml
in
temp_xml
:
if
xml
.
endswith
(
".xml"
):
total_xml
.
append
(
xml
)
num
=
len
(
total_xml
)
list
=
range
(
num
)
tv
=
int
(
num
*
trainval_percent
)
tr
=
int
(
tv
*
train_percent
)
trainval
=
random
.
sample
(
list
,
tv
)
train
=
random
.
sample
(
trainval
,
tr
)
print
(
"train and val size"
,
tv
)
print
(
"traub suze"
,
tr
)
ftrainval
=
open
(
os
.
path
.
join
(
saveBasePath
,
'trainval.txt'
),
'w'
)
ftest
=
open
(
os
.
path
.
join
(
saveBasePath
,
'test.txt'
),
'w'
)
ftrain
=
open
(
os
.
path
.
join
(
saveBasePath
,
'train.txt'
),
'w'
)
fval
=
open
(
os
.
path
.
join
(
saveBasePath
,
'val.txt'
),
'w'
)
for
i
in
list
:
name
=
total_xml
[
i
][:
-
4
]
+
'
\n
'
if
i
in
trainval
:
ftrainval
.
write
(
name
)
if
i
in
train
:
ftrain
.
write
(
name
)
else
:
fval
.
write
(
name
)
else
:
ftest
.
write
(
name
)
ftrainval
.
close
()
ftrain
.
close
()
fval
.
close
()
ftest
.
close
()
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录