SVM从零开始 (五) 写训练集train.txt, 测试集test.txt 的代码

写训练集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