{ "cells": [ { "cell_type": "markdown", "metadata": {}, "source": [ "## Example Inference" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Set Up\n", "prerequisites\n", "- python 3.12/11\n", "\n", "``` shell\n", "$ cd ./package/{platform}/\n", "$ pip install KneronPLUS-{version}-py3-none-any.whl\n", "$ pip install opencv-python\n", "\n", "```" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Connect KL520" ] }, { "cell_type": "code", "execution_count": 1, "metadata": {}, "outputs": [], "source": [ "import cv2\n", "import kp" ] }, { "cell_type": "code", "execution_count": 2, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "13\n" ] } ], "source": [ "device_descriptors = kp.core.scan_devices()\n", "\n", "if 0 < device_descriptors.device_descriptor_number:\n", " usb_port_id = device_descriptors.device_descriptor_list[0].usb_port_id\n", " print(usb_port_id)\n", "else:\n", " print('Error: no Kneron device connect.')\n", " exit(0)" ] }, { "cell_type": "code", "execution_count": 3, "metadata": {}, "outputs": [], "source": [ "device_group = kp.core.connect_devices(usb_port_ids=[22])" ] }, { "cell_type": "code", "execution_count": 4, "metadata": {}, "outputs": [], "source": [ "kp.core.set_timeout(device_group=device_group,\n", " milliseconds=5000)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### Load firmware, model, and test image" ] }, { "cell_type": "code", "execution_count": 5, "metadata": {}, "outputs": [], "source": [ "SCPU_FW_PATH = '../../res/firmware/KL520/fw_scpu.bin'\n", "NCPU_FW_PATH = '../../res/firmware/KL520/fw_ncpu.bin'\n", "kp.core.load_firmware_from_file(device_group=device_group,\n", " scpu_fw_path=SCPU_FW_PATH,\n", " ncpu_fw_path=NCPU_FW_PATH)" ] }, { "cell_type": "code", "execution_count": 6, "metadata": {}, "outputs": [], "source": [ "MODEL_FILE_PATH = './declutter_520.nef'\n", "model_nef_descriptor = kp.core.load_model_from_file(device_group=device_group,\n", " file_path=MODEL_FILE_PATH)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### pre-processing" ] }, { "cell_type": "code", "execution_count": 7, "metadata": {}, "outputs": [], "source": [ "IMAGE_FILE_PATH = './test.jpg'\n", "\n", "img = cv2.imread(filename=IMAGE_FILE_PATH)\n", "img_bgr565 = cv2.cvtColor(src=img, code=cv2.COLOR_BGR2BGR565)" ] }, { "cell_type": "code", "execution_count": 8, "metadata": {}, "outputs": [], "source": [ "generic_inference_input_descriptor = kp.GenericImageInferenceDescriptor(\n", " model_id=model_nef_descriptor.models[0].id,\n", " inference_number=0,\n", " input_node_image_list=[\n", " kp.GenericInputNodeImage(\n", " image=img_bgr565,\n", " image_format=kp.ImageFormat.KP_IMAGE_FORMAT_RGB565,\n", " resize_mode=kp.ResizeMode.KP_RESIZE_ENABLE,\n", " padding_mode=kp.PaddingMode.KP_PADDING_CORNER,\n", " normalize_mode=kp.NormalizeMode.KP_NORMALIZE_KNERON\n", " )\n", " ]\n", ")" ] }, { "cell_type": "code", "execution_count": 9, "metadata": {}, "outputs": [], "source": [ "kp.inference.generic_image_inference_send(device_group=device_group,\n", " generic_inference_input_descriptor=generic_inference_input_descriptor)\n", "generic_raw_result = kp.inference.generic_image_inference_receive(device_group=device_group)" ] }, { "cell_type": "code", "execution_count": 10, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "[{\n", " \"width\": 1,\n", " \"height\": 1,\n", " \"channel\": 512,\n", " \"channels_ordering\": \"ChannelOrdering.KP_CHANNEL_ORDERING_CHW\",\n", " \"num_data\": 512,\n", " \"ndarray\": [\n", " \"[[[[0.5060123 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[1.113227 ]]\",\n", " \"\",\n", " \" [[1.1638283 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[2.3782578 ]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[1.4674357 ]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[3.036074 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[3.4408836 ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[1.2650307 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[2.3782578 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[1.0626259 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[2.6818652 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[1.6698406 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[2.1252518 ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[1.2144295 ]]\",\n", " \"\",\n", " \" [[0.91082215]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.8602209 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[1.2650307 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[1.315632 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.8602209 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[1.0626259 ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.9614234 ]]\",\n", " \"\",\n", " \" [[0.9614234 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[1.2650307 ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[2.3276565 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[2.226454 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[1.3662332 ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.91082215]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.8602209 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[3.1372762 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[2.4794602 ]]\",\n", " \"\",\n", " \" [[1.9228468 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[1.518037 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[1.0626259 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[1.0120246 ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.91082215]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[3.4914849 ]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[1.7204418 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.8602209 ]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.70841724]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.8602209 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[1.0120246 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[1.7710431 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.7590185 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[1.518037 ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.60721475]]\",\n", " \"\",\n", " \" [[0.657816 ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.45541108]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[1.113227 ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.25300616]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[0.15180369]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.35420862]]\",\n", " \"\",\n", " \" [[0.10120246]]\",\n", " \"\",\n", " \" [[1.315632 ]]\",\n", " \"\",\n", " \" [[0.30360737]]\",\n", " \"\",\n", " \" [[0.5060123 ]]\",\n", " \"\",\n", " \" [[0.40480983]]\",\n", " \"\",\n", " \" [[0.5566135 ]]\",\n", " \"\",\n", " \" [[0.20240492]]\",\n", " \"\",\n", " \" [[0. ]]\",\n", " \"\",\n", " \" [[0.05060123]]\",\n", " \"\",\n", " \" [[1.7204418 ]]\",\n", " \"\",\n", " \" [[0.80961967]]\",\n", " \"\",\n", " \" [[0.05060123]]]]\"\n", " ]\n", "}]\n" ] } ], "source": [ "inf_node_output_list = []\n", "\n", "for node_idx in range(generic_raw_result.header.num_output_node):\n", " inference_float_node_output = kp.inference.generic_inference_retrieve_float_node(node_idx=node_idx,\n", " generic_raw_result=generic_raw_result,\n", " channels_ordering=kp.ChannelOrdering.KP_CHANNEL_ORDERING_CHW)\n", " inf_node_output_list.append(inference_float_node_output)\n", "\n", "print(inf_node_output_list)" ] }, { "cell_type": "markdown", "metadata": {}, "source": [ "### post-processing" ] }, { "cell_type": "code", "execution_count": 22, "metadata": {}, "outputs": [ { "name": "stdout", "output_type": "stream", "text": [ "dimension of feature tensor: 512\n", "\n", "[0.5060123 0. 0.30360737 0.5060123 0.25300616 0.\n", " 0.05060123 1.113227 1.1638283 0.05060123 0.30360737 0.10120246\n", " 0.5060123 0.20240492 0.05060123 0.5566135 0.05060123 0.25300616\n", " 0.25300616 0.10120246 0.05060123 0.7590185 0.5566135 0.\n", " 0.05060123 0.15180369 0.30360737 0.7590185 0.657816 0.05060123\n", " 0. 0.5566135 0.05060123 0.25300616 0. 0.05060123\n", " 0.35420862 0.05060123 2.3782578 0.60721475 0.10120246 0.10120246\n", " 0.30360737 0.70841724 0.45541108 0.35420862 0.25300616 0.30360737\n", " 0.25300616 0.05060123 0.657816 0.05060123 0.45541108 0.10120246\n", " 0.15180369 0.15180369 0.657816 0.25300616 0. 0.10120246\n", " 0. 0.20240492 0.20240492 0.40480983 0.80961967 0.25300616\n", " 0.25300616 0.05060123 0.05060123 0. 0.05060123 0.45541108\n", " 0.45541108 0.40480983 0.10120246 0. 1.4674357 0.60721475\n", " 0.10120246 0.15180369 0.15180369 0.7590185 0.45541108 0.\n", " 0. 0.35420862 0.05060123 0.5566135 0.15180369 0.10120246\n", " 0.05060123 0.40480983 0. 0.30360737 0.25300616 0.\n", " 3.036074 0.15180369 0.05060123 0. 0.10120246 0.\n", " 0. 0. 3.4408836 0.5566135 0.05060123 0.\n", " 0.10120246 0.30360737 0.40480983 0.5060123 0.15180369 0.\n", " 0. 0.05060123 0.5060123 0.25300616 0.05060123 0.25300616\n", " 1.2650307 0.5060123 0.30360737 0.5566135 0. 0.25300616\n", " 0.20240492 0.30360737 2.3782578 0. 0.20240492 0.35420862\n", " 0. 0. 0. 0.7590185 0. 0.70841724\n", " 0. 0.15180369 1.0626259 0.25300616 0.80961967 0.5060123\n", " 0.05060123 0.05060123 0.05060123 2.6818652 0.5060123 0.20240492\n", " 0.657816 0.20240492 0.5060123 0.05060123 0.35420862 0.20240492\n", " 0.05060123 0.30360737 0.10120246 0.25300616 0.30360737 0.\n", " 0.15180369 0.80961967 0.05060123 0.30360737 0. 0.\n", " 0.45541108 0.20240492 0. 0.30360737 0. 0.\n", " 0.7590185 0.5566135 0.10120246 0. 0.35420862 0.\n", " 0.10120246 0.30360737 1.6698406 0.20240492 2.1252518 0.45541108\n", " 0.30360737 0.5060123 0.15180369 0.35420862 0.20240492 0.20240492\n", " 1.2144295 0.91082215 0.05060123 0. 0.20240492 0.80961967\n", " 0.8602209 0.25300616 0.05060123 0.30360737 0.80961967 0.25300616\n", " 0. 0.10120246 0.45541108 0.35420862 1.2650307 0.25300616\n", " 0. 0.05060123 0.35420862 0.45541108 0.15180369 0.40480983\n", " 1.315632 0.5060123 0.20240492 0.70841724 0.05060123 0.35420862\n", " 0.20240492 0.60721475 0.30360737 0.20240492 0.8602209 0.05060123\n", " 0.05060123 0.45541108 0.20240492 0. 0.40480983 0.10120246\n", " 0.7590185 0.05060123 1.0626259 0.35420862 0. 0.10120246\n", " 0.05060123 0.25300616 0.9614234 0.9614234 0.10120246 0.10120246\n", " 0.5566135 0.20240492 0.10120246 1.2650307 0.45541108 0.20240492\n", " 0.20240492 0.05060123 0.45541108 0.60721475 0.5060123 0.\n", " 0.40480983 0. 0.25300616 0.40480983 0.70841724 0.45541108\n", " 0.35420862 2.3276565 0. 0.657816 0.25300616 0.35420862\n", " 0.05060123 0.5060123 0.5060123 2.226454 0.10120246 0.20240492\n", " 0.70841724 0.45541108 0.10120246 0.20240492 0.5060123 0.10120246\n", " 0.45541108 0.10120246 0. 0.657816 0.10120246 0.25300616\n", " 0.25300616 0.10120246 0.40480983 0. 0.5060123 0.35420862\n", " 0. 0.40480983 0.20240492 0. 0.15180369 0.15180369\n", " 1.3662332 0.35420862 0.10120246 0.20240492 0.91082215 0.\n", " 0. 0.8602209 0.05060123 0. 0.5060123 0.7590185\n", " 0.05060123 0.05060123 0.25300616 0. 0. 0.\n", " 0.15180369 0.25300616 0.20240492 0.05060123 0. 0.10120246\n", " 0.30360737 0.05060123 0. 0. 0.25300616 0.\n", " 0.657816 0. 0.30360737 0.20240492 0.20240492 0.20240492\n", " 3.1372762 0. 0. 0.05060123 0.30360737 0.5566135\n", " 0.5060123 2.4794602 1.9228468 0.10120246 0.05060123 0.7590185\n", " 0.80961967 0.40480983 0.60721475 0.05060123 0.5566135 0.\n", " 0.25300616 0.5060123 0.10120246 0.20240492 0.40480983 0.20240492\n", " 0.60721475 0.05060123 0.35420862 0. 1.518037 0.10120246\n", " 0.60721475 0.35420862 1.0626259 0. 0.05060123 0.\n", " 0. 0. 1.0120246 0.5060123 0.91082215 0.80961967\n", " 0.5060123 0.15180369 0.05060123 0. 3.4914849 0.60721475\n", " 0.657816 0. 0.10120246 0.30360737 0. 0.5060123\n", " 0.70841724 0.80961967 0.5566135 0.60721475 0.05060123 0.20240492\n", " 0.30360737 0.657816 0.40480983 0. 0.35420862 0.20240492\n", " 0.15180369 0.60721475 1.7204418 0.10120246 0.35420862 0.40480983\n", " 0.10120246 0. 0.30360737 0.8602209 0.60721475 0.70841724\n", " 0.10120246 0.15180369 0.657816 0.8602209 0.25300616 0.\n", " 0.40480983 0.30360737 0. 0.45541108 1.0120246 0.\n", " 0.5566135 0.5566135 0.25300616 0.30360737 0. 0.80961967\n", " 0.35420862 0.05060123 0.10120246 1.7710431 0.10120246 0.10120246\n", " 0.60721475 0.7590185 0.10120246 0.40480983 0.25300616 0.657816\n", " 0.7590185 0.25300616 0.25300616 0.05060123 0.05060123 0.\n", " 0.5060123 0. 0.10120246 0.25300616 0.5060123 0.657816\n", " 0.30360737 0.05060123 0.25300616 1.518037 0.15180369 0.25300616\n", " 0.45541108 0. 0.45541108 0.60721475 0.657816 0.\n", " 0.45541108 0.5060123 0.40480983 0.05060123 0. 0.15180369\n", " 0.20240492 0.5060123 0.10120246 0.30360737 0.5060123 0.25300616\n", " 0.5060123 0.45541108 0.5566135 0.05060123 1.113227 0.10120246\n", " 0.05060123 0.25300616 0.35420862 0. 0.10120246 0.\n", " 0. 0.10120246 0.25300616 0. 0.10120246 0.15180369\n", " 0. 0.35420862 0.10120246 1.315632 0.30360737 0.5060123\n", " 0.40480983 0.5566135 0.20240492 0. 0.05060123 1.7204418\n", " 0.80961967 0.05060123]\n" ] } ], "source": [ "import numpy as np\n", "import re\n", "\n", "data = inf_node_output_list[0]\n", "\n", "raw_ndarray = data.ndarray\n", "\n", "if isinstance(raw_ndarray, np.ndarray):\n", " ndarray_np = raw_ndarray\n", "else:\n", " ndarray_np = np.array(raw_ndarray)\n", "\n", "ndarray_np = ndarray_np.flatten() \n", "# ndarray_np = ndarray_np.reshape((data.channel, data.height, data.width))\n", "\n", "print(f\"dimension of feature tensor: {ndarray_np.size}\\n\")\n", "print(ndarray_np)" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.12.6" } }, "nbformat": 4, "nbformat_minor": 2 }