prototxt切片

import tensorflow as tf
import numpy as np
fname = './b.prototxt'
part_prototxt_file = './'
all_index_ = './all_index.txt'#新建一个文件夹
var_txt_ = 'var.txt'
part_name = part_prototxt_file +'ccc.prototxt'
part_typer = 'the_typer'
var_txt = open(var_txt_,'r')
all_prototxt = open(fname,'r')
all_index = open(all_index_, 'w')
part_prototxt = []
while True: 
    lines = all_prototxt.readline() # 整行读取数据
    if not lines:
        f = open(part_prototxt_file+'end.prototxt', 'w')
        for i in range(len(part_prototxt)):
            f.write(str(part_prototxt[i]))   
        f.close()
        part_prototxt.clear()
        break   
    find_name = lines.strip()
    if find_name[:4] == 'name':
        part_name = find_name[5:]+'.prototxt'
    if find_name[:5] == 'typer':
        part_typer = find_name[6:]   
    if find_name == "###****":
        print(part_typer)
        f = open(part_prototxt_file+part_name, 'w')
        for i in range(len(part_prototxt)):
            f.write(str(part_prototxt[i]))   
        f.close()
        part_prototxt.clear()
        all_index.write(part_name +'\n')
        if part_typer == 'convlustion' or part_typer == 'sc':
            verline = var_txt.readline() # 整行读取数据
            all_index.write(verline)
        if part_typer == 'bn'or part_typer == 'fc':
            verline = var_txt.readline() # 整行读取数据
            all_index.write(verline)
            verline = var_txt.readline() # 整行读取数据
            all_index.write(verline)
    else:
        part_prototxt.append(lines)
all_index.close()
var_txt.close()
all_prototxt.close()