build multiple suffixes and socs at once

conv.py

@@ -1,9 +1,12 @@
 from rknn.api import RKNN
+import os
 
-INPUT_MODEL = "yolov8x.onnx"
+for suffix in ["n", "s", "m", "l", "x"]:
+    for soc in ["rk3562","rk3566", "rk3568", "rk3588"]:
+        INPUT_MODEL = 'yolov8{}.onnx'.format(suffix)
         WIDTH = 320
         HEIGHT = 320
-OUTPUT_MODEL_BASENAME = 'yolov8x'
+        OUTPUT_MODEL_BASENAME = 'yolov8{}'.format(suffix)
         QUANTIZATION = False
         DATASET = './dataset_coco10.txt'
 
@@ -16,7 +19,7 @@ QUANTIZED_ALGORITHM = "normal"
         QUANTIZED_METHOD = "channel"
         FLOAT_DTYPE = "float16"
         OPTIMIZATION_LEVEL = 2
-TARGET_PLATFORM = "rk3588"
+        TARGET_PLATFORM = soc
         CUSTOM_STRING = None
         REMOVE_WEIGHT = None
         COMPRESS_WEIGHT = False
@@ -25,8 +28,8 @@ MODEL_PRUNNING = False
         OP_TARGET = None
         DYNAMIC_INPUT = None
 
-
-OUTPUT_MODEL = OUTPUT_MODEL_BASENAME + '-' + str(WIDTH) + 'x' + str(HEIGHT) + ".rknn"
+        OUTPUT_MODEL_FILE = "./output/{}/{}-{}x{}-{}.rknn".format(soc, OUTPUT_MODEL_BASENAME, WIDTH, HEIGHT, soc)
+        os.makedirs("./output/{}".format(soc), exist_ok=True)
 
         rknn = RKNN()
         rknn.config(mean_values=MEAN_VALUES,
@@ -55,6 +58,6 @@ if rknn.build(do_quantization=QUANTIZATION, dataset=DATASET) != 0:
             print('Error building model.')
             exit()
 
-if rknn.export_rknn("./output/" + OUTPUT_MODEL) != 0:
+        if rknn.export_rknn(OUTPUT_MODEL_FILE) != 0:
             print('Error exporting rknn model.')
             exit()