def print_args(print_fn=my_log): for k, v in args._get_kwargs(): print_fn('{} = {}'.format(k, v)) print_fn('')
import torch from flask import Flask, request from utils.json_utils import jsonify from model import initital_model, class_id2name from transform import transform_image import time from collections import OrderedDict import codecs from args import args # 获取所有配置参数 state = {k: v for k, v in args._get_kwargs()} print("state = ", state) app = Flask(__name__) # 设置编码-否则返回数据中文时候-乱码 app.config['JSON_AS_ASCII'] = False # 加载Label2Name Mapping class_id2name = {} for line in codecs.open('data/garbage_label.txt', 'r', encoding='utf-8'): line = line.strip() _id = line.split(":")[0] _name = line.split(":")[1] class_id2name[int(_id)] = _name device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # 设备 print('Pytorch garbage-classification Serving on {} ...'.format(device)) num_classes = len(class_id2name) model_name = args.model_name model_path = args.resume # --resume checkpoint/garbage_resnext101_model_2_1111_4211.pth print("model_name = ", model_name) print("model_path = ", model_path)
def print_args(): for k, v in args._get_kwargs(): my_log('{} = {}'.format(k, v)) my_log('')