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('')
