mmdnn/models/GenerateMdByDataset.py [5:38]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
markdown_code = str()

framework_list = ['caffe', 'cntk', 'coreml', 'darknet', 'mxnet', 'pytorch', 'tensorflow']  # Haven't added 'keras' yet
frame_model_map = {
     'caffe': {'architecture':'prototxt', 'weights':'caffemodel'},
     'cntk': {'architecture':'model'},
     'coreml': {'architecture':'mlmodel'},
     'darknet': {'architecture':'cfg', 'weights':'weights'},
     'mxnet': {'architecture':'json', 'weights':'params'},
     'pytorch': {'architecture':'pth'},
     'tensorflow': {'architecture':'tgz'}
}  # Haven't add 'keras' yet
dataset_list = ['imagenet', 'imagenet11k', 'Pascal VOC', 'grocery100']

def add_code(code):
    global markdown_code
    markdown_code += code

def add_header(level, code):
    add_code("#" * level + " " + code + '\n\n')

def draw_line(num):
    add_code("| " * num + "|\n")
    add_code(("|-" * num + "|\n"))

def save_code(filepath):
    with open(filepath, 'w') as f:
        f.write(markdown_code)
    print("Markdown generate succeeded!")

def LoadJson(json_path):
    with open(json_path, encoding='utf-8') as f:
        data = json.load(f)
    return data
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



mmdnn/models/GenerateMdFromJson.py [5:38]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
markdown_code = str()

framework_list = ['caffe', 'cntk', 'coreml', 'darknet', 'mxnet', 'pytorch', 'tensorflow']  # Haven't add 'keras' yet
frame_model_map = {
     'caffe': {'architecture':'prototxt', 'weights':'caffemodel'},
     'cntk': {'architecture':'model'},
     'coreml': {'architecture':'mlmodel'},
     'darknet': {'architecture':'cfg', 'weights':'weights'},
     'mxnet': {'architecture':'json', 'weights':'params'},
     'pytorch': {'architecture':'pth'},
     'tensorflow': {'architecture':'tgz'}
}  # Haven't add 'keras' yet
dataset_list = ['imagenet', 'imagenet11k', 'Pascal VOC', 'grocery100']

def add_code(code):
    global markdown_code
    markdown_code += code

def add_header(level, code):
    add_code("#" * level + " " + code + '\n\n')

def draw_line(num):
    add_code("| " * num + "|\n")
    add_code(("|-" * num + "|\n"))

def save_code(filepath):
    with open(filepath, 'w') as f:
        f.write(markdown_code)
    print("Markdown generate succeeded!")

def LoadJson(json_path):
    with open(json_path, encoding='utf-8') as f:
        data = json.load(f)
    return data
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -