增加学习率调度方式

tools/model_rknn_transform.py

@@ -6,15 +6,14 @@ import time
 import sys
 import numpy as np
 import cv2
-from config import config as conf
 from rknn.api import RKNN
-
-import config
-
+import yaml
+with open('../configs/transform.yml', 'r') as f:
+    conf = yaml.load(f, Loader=yaml.FullLoader)
 # ONNX_MODEL = 'resnet50v2.onnx'
 # RKNN_MODEL = 'resnet50v2.rknn'
-ONNX_MODEL = 'checkpoints/resnet18_scale=1.0/best.onnx'
-RKNN_MODEL = 'checkpoints/resnet18_scale=1.0/best.rknn'
+ONNX_MODEL = conf['models']['onnx_model']
+RKNN_MODEL = conf['models']['rknn_model']
 
 
 # ONNX_MODEL = 'v3_small_0424.onnx'