写训练集train.txt, 测试集test.txt 的代码
辅助工具 => Change_Name.py, Write_Label_txt.py
Write_Label_txt.py
0 完整代码
if __name__ == '__main__':
'''
=> 用于完成test.txt, 要在此文件目录下事先创建test.txt文件
与后面完成train.txt分开单独使用时, 直接注释掉另一个
'''
# /=== ===> test.txt 写入pic_name & label <=== ===\
with open('test.txt', 'w') as f: # train.txt或者test.txt
for i in range(30):
i = i+1 # 0-29 => 1-30
x = '%03d' % i
f.writelines('1-' + str(x) + '.jpg' + ' ' + '1' + '\n')
'''
结果如下:
1-001.jpg 1
1-002.jpg 1
1-003.jpg 1
1-004.jpg 1
1-005.jpg 1
1-006.jpg 1
1-007.jpg 1
1-008.jpg 1
1-009.jpg 1
1-010.jpg 1
1-011.jpg 1
1-012.jpg 1
1-013.jpg 1
1-014.jpg 1
1-015.jpg 1
1-016.jpg 1
1-017.jpg 1
1-018.jpg 1
1-019.jpg 1
1-020.jpg 1
1-021.jpg 1
1-022.jpg 1
1-023.jpg 1
1-024.jpg 1
1-025.jpg 1
1-026.jpg 1
1-027.jpg 1
1-028.jpg 1
1-029.jpg 1
1-030.jpg 1
'''
# \=== ===> test.txt 写入pic_name & label <=== ===/
'''
=> 用于完成train.txt, 要在此文件目录下事先创建train.txt文件
与前面完成test.txt分开单独使用时, 直接注释掉另一个即可
'''
# /=== ===> train.txt 写入pic_name & label <=== ===\
with open('train.txt', 'w') as f: # train.txt或者test.txt
for i in range(215):
i = i+1 # 0-214 => 1-215
i = i+30 # 1-215 => 31-245
x = '%03d' % i
f.writelines('1-' + str(x) + '.jpg' + ' ' + '1' + '\n')
for i in range(162):
i = i+1 # 0-161 => 1-162
x = '%03d' % i
f.writelines('2-' + str(x) + '.jpg' + ' ' + '2' + '\n')
for i in range(308):
i = i+1 # 0-307 => 1-308
x = '%03d' % i
f.writelines('3-' + str(x) + '.jpg' + ' ' + '3' + '\n')
for i in range(337):
i = i + 1 # 0-336 => 1-337
x = '%03d' % i
f.writelines('4-' + str(x) + '.jpg' + ' ' + '4' + '\n')
'''
1-031.jpg 1 =>1 start
1-032.jpg 1
...
1-244.jpg 1
1-245.jpg 1 =>1 end
2-001.jpg 2 =>2 start
2-002.jpg 2
...
2-161.jpg 2
2-162.jpg 2 =>2 end
3-002.jpg 3 =>3 start
3-003.jpg 3
...
3-307.jpg 3
3-308.jpg 3 =>3 end
4-001.jpg 4 =>4 start
4-002.jpg 4
...
4-336.jpg 4
4-337.jpg 4 =>4 end
'''
# \=== ===> train.txt 写入pic_name & label <=== ===/
1 test.txt为例:
'''
=> 用于完成test.txt, 要在此文件目录下事先创建test.txt文件
与后面完成train.txt分开单独使用时, 直接注释掉另一个
'''
# /=== ===> test.txt 写入pic_name & label <=== ===\
with open('test.txt', 'w') as f: # train.txt或者test.txt
for i in range(30):
i = i+1 # 0-29 => 1-30
x = '%03d' % i
f.writelines('1-' + str(x) + '.jpg' + ' ' + '1' + '\n')
'''
结果如下:
1-001.jpg 1
1-002.jpg 1
1-003.jpg 1
1-004.jpg 1
1-005.jpg 1
1-006.jpg 1
1-007.jpg 1
1-008.jpg 1
1-009.jpg 1
1-010.jpg 1
1-011.jpg 1
1-012.jpg 1
1-013.jpg 1
1-014.jpg 1
1-015.jpg 1
1-016.jpg 1
1-017.jpg 1
1-018.jpg 1
1-019.jpg 1
1-020.jpg 1
1-021.jpg 1
1-022.jpg 1
1-023.jpg 1
1-024.jpg 1
1-025.jpg 1
1-026.jpg 1
1-027.jpg 1
1-028.jpg 1
1-029.jpg 1
1-030.jpg 1
'''
# \=== ===> test.txt 写入pic_name & label <=== ===/
test.txt为:
2 train.txt:
# \=== ===> test.txt 写入pic_name & label <=== ===/
'''
=> 用于完成train.txt, 要在此文件目录下事先创建train.txt文件
与前面完成test.txt分开单独使用时, 直接注释掉另一个即可
'''
# /=== ===> train.txt 写入pic_name & label <=== ===\
with open('train.txt', 'w') as f: # train.txt或者test.txt
for i in range(215):
i = i+1 # 0-214 => 1-215
i = i+30 # 1-215 => 31-245
x = '%03d' % i
f.writelines('1-' + str(x) + '.jpg' + ' ' + '1' + '\n')
for i in range(162):
i = i+1 # 0-161 => 1-162
x = '%03d' % i
f.writelines('2-' + str(x) + '.jpg' + ' ' + '2' + '\n')
for i in range(308):
i = i+1 # 0-307 => 1-308
x = '%03d' % i
f.writelines('3-' + str(x) + '.jpg' + ' ' + '3' + '\n')
for i in range(337):
i = i + 1 # 0-336 => 1-337
x = '%03d' % i
f.writelines('4-' + str(x) + '.jpg' + ' ' + '4' + '\n')
'''
1-031.jpg 1 =>1 start
1-032.jpg 1
...
1-244.jpg 1
1-245.jpg 1 =>1 end
2-001.jpg 2 =>2 start
2-002.jpg 2
...
2-161.jpg 2
2-162.jpg 2 =>2 end
3-002.jpg 3 =>3 start
3-003.jpg 3
...
3-307.jpg 3
3-308.jpg 3 =>3 end
4-001.jpg 4 =>4 start
4-002.jpg 4
...
4-336.jpg 4
4-337.jpg 4 =>4 end
'''
# \=== ===> train.txt 写入pic_name & label <=== ===/
1:tbj的前面30张被用来测试了
下面是train.txt 开头部分
1-031.jpg 1
...
1-245.jpg 1
2-001.jpg 2
...
2-162.jpg 2
3-001.jpg 3
...
3-308.jpg 3
4-001.jpg 4
...
4-337.jpg 4