model-meta-data.ipynb

{"nbformat":4,"nbformat_minor":0,"metadata":{"colab":{"provenance":[],"collapsed_sections":[],"authorship_tag":"ABX9TyP6E3jRYuIrDHwVXAX01msg"},"kernelspec":{"name":"python3","display_name":"Python 3"},"language_info":{"name":"python"}},"cells":[{"cell_type":"code","source":["from google.colab import drive \n","drive.mount('/content/drive')"],"metadata":{"colab":{"base_uri":"https://localhost:8080/"},"id":"iEqRxI0ortCV","executionInfo":{"status":"ok","timestamp":1667931116885,"user_tz":-540,"elapsed":3964,"user":{"displayName":"한지희","userId":"02830074421427224530"}},"outputId":"91c8abc5-ca13-4073-eb9d-03113900b6b0"},"execution_count":null,"outputs":[{"output_type":"stream","name":"stdout","text":["Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount(\"/content/drive\", force_remount=True).\n"]}]},{"cell_type":"code","source":["!pip install tflite-support"],"metadata":{"colab":{"base_uri":"https://localhost:8080/","height":667},"id":"r2_VKQ5Nt6vn","executionInfo":{"status":"ok","timestamp":1667931049602,"user_tz":-540,"elapsed":16763,"user":{"displayName":"한지희","userId":"02830074421427224530"}},"outputId":"7abfa22c-af28-4320-af13-463d91f33554"},"execution_count":null,"outputs":[{"output_type":"stream","name":"stdout","text":["Looking in indexes: https://pypi.org/simple, https://us-python.pkg.dev/colab-wheels/public/simple/\n","Collecting tflite-support\n","  Downloading tflite_support-0.4.3-cp37-cp37m-manylinux2014_x86_64.whl (60.9 MB)\n","\u001b[K     |████████████████████████████████| 60.9 MB 1.2 MB/s \n","\u001b[?25hCollecting flatbuffers>=2.0\n","  Downloading flatbuffers-22.10.26-py2.py3-none-any.whl (26 kB)\n","Collecting sounddevice>=0.4.4\n","  Downloading sounddevice-0.4.5-py3-none-any.whl (31 kB)\n","Collecting pybind11>=2.6.0\n","  Downloading pybind11-2.10.1-py3-none-any.whl (216 kB)\n","\u001b[K     |████████████████████████████████| 216 kB 69.7 MB/s \n","\u001b[?25hRequirement already satisfied: absl-py>=0.7.0 in /usr/local/lib/python3.7/dist-packages (from tflite-support) (1.3.0)\n","Requirement already satisfied: numpy>=1.20.0 in /usr/local/lib/python3.7/dist-packages (from tflite-support) (1.21.6)\n","Collecting protobuf<4,>=3.18.0\n","  Downloading protobuf-3.20.3-cp37-cp37m-manylinux_2_5_x86_64.manylinux1_x86_64.whl (1.0 MB)\n","\u001b[K     |████████████████████████████████| 1.0 MB 43.4 MB/s \n","\u001b[?25hRequirement already satisfied: CFFI>=1.0 in /usr/local/lib/python3.7/dist-packages (from sounddevice>=0.4.4->tflite-support) (1.15.1)\n","Requirement already satisfied: pycparser in /usr/local/lib/python3.7/dist-packages (from CFFI>=1.0->sounddevice>=0.4.4->tflite-support) (2.21)\n","Installing collected packages: sounddevice, pybind11, protobuf, flatbuffers, tflite-support\n","  Attempting uninstall: protobuf\n","    Found existing installation: protobuf 3.17.3\n","    Uninstalling protobuf-3.17.3:\n","      Successfully uninstalled protobuf-3.17.3\n","  Attempting uninstall: flatbuffers\n","    Found existing installation: flatbuffers 1.12\n","    Uninstalling flatbuffers-1.12:\n","      Successfully uninstalled flatbuffers-1.12\n","\u001b[31mERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.\n","tensorflow 2.9.2 requires flatbuffers<2,>=1.12, but you have flatbuffers 22.10.26 which is incompatible.\n","tensorflow 2.9.2 requires protobuf<3.20,>=3.9.2, but you have protobuf 3.20.3 which is incompatible.\n","tensorboard 2.9.1 requires protobuf<3.20,>=3.9.2, but you have protobuf 3.20.3 which is incompatible.\u001b[0m\n","Successfully installed flatbuffers-22.10.26 protobuf-3.20.3 pybind11-2.10.1 sounddevice-0.4.5 tflite-support-0.4.3\n"]},{"output_type":"display_data","data":{"application/vnd.colab-display-data+json":{"pip_warning":{"packages":["google"]}}},"metadata":{}}]},{"cell_type":"code","execution_count":null,"metadata":{"id":"UjHPzWZLpDDk"},"outputs":[],"source":["from tflite_support import flatbuffers\n","from tflite_support import metadata as _metadata\n","from tflite_support import metadata_schema_py_generated as _metadata_fb\n","\n","from tflite_support import flatbuffers\n","from tflite_support import metadata as _metadata\n","from tflite_support import metadata_schema_py_generated as _metadata_fb\n","\n","model_meta = _metadata_fb.ModelMetadataT()\n","model_meta.name = \"AngDroid Claasification Model\"\n","model_meta.description = (\"AngDroid Fruits Classification\")\n","model_meta.version = \"v1\"\n","model_meta.author = \"jihee.dev\"\n","model_meta.license = (\"Apache License. Version 2.0 \"\n","                      \"http://www.apache.org/licenses/LICENSE-2.0.\")"]},{"cell_type":"code","source":["# Creates input info.\n","input_meta = _metadata_fb.TensorMetadataT()\n","\n","# Creates output info.\n","output_meta = _metadata_fb.TensorMetadataT()"],"metadata":{"id":"kuEBXCY6cW6A"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["input_meta.name = \"image\"\n","input_meta.description = (\n","    \"Input image to be classified. The expected image is {0} x {1}, with \"\n","    \"three channels (red, blue, and green) per pixel. Each value in the \"\n","    \"tensor is a single byte between 0 and 255.\".format(160, 160))\n","input_meta.content = _metadata_fb.ContentT()\n","input_meta.content.contentProperties = _metadata_fb.ImagePropertiesT()\n","input_meta.content.contentProperties.colorSpace = (\n","    _metadata_fb.ColorSpaceType.RGB)\n","input_meta.content.contentPropertiesType = (\n","    _metadata_fb.ContentProperties.ImageProperties)\n","input_normalization = _metadata_fb.ProcessUnitT()\n","input_normalization.optionsType = (\n","    _metadata_fb.ProcessUnitOptions.NormalizationOptions)\n","input_normalization.options = _metadata_fb.NormalizationOptionsT()\n","input_normalization.options.mean = [127.5]\n","input_normalization.options.std = [127.5]\n","input_meta.processUnits = [input_normalization]\n","input_stats = _metadata_fb.StatsT()\n","input_stats.max = [255]\n","input_stats.min = [0]\n","input_meta.stats = input_stats"],"metadata":{"id":"hrGwO1Sjdl30"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["import os\n","\n","output_meta = _metadata_fb.TensorMetadataT()\n","output_meta.name = \"fruits class\"\n","output_meta.description = \"Probabilities of the 1001 labels respectively.\"\n","output_meta.content = _metadata_fb.ContentT()\n","output_meta.content.content_properties = _metadata_fb.FeaturePropertiesT()\n","output_meta.content.contentPropertiesType = (\n","    _metadata_fb.ContentProperties.FeatureProperties)\n","output_stats = _metadata_fb.StatsT()\n","output_stats.max = [1.0]\n","output_stats.min = [0.0]\n","output_meta.stats = output_stats\n","label_file = _metadata_fb.AssociatedFileT()\n","label_file.name = os.path.basename(\"/content/drive/MyDrive/Colab Notebooks/labels.txt\")\n","label_file.description = \"Labels for objects that the model can recognize.\"\n","label_file.type = _metadata_fb.AssociatedFileType.TENSOR_AXIS_LABELS\n","output_meta.associatedFiles = [label_file]"],"metadata":{"id":"NWn_t0S5d3I8"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["# Creates subgraph info.\n","subgraph = _metadata_fb.SubGraphMetadataT()\n","subgraph.inputTensorMetadata = [input_meta]\n","subgraph.outputTensorMetadata = [output_meta]\n","model_meta.subgraphMetadata = [subgraph]\n","\n","b = flatbuffers.Builder(0)\n","b.Finish(\n","    model_meta.Pack(b),\n","    _metadata.MetadataPopulator.METADATA_FILE_IDENTIFIER)\n","metadata_buf = b.Output()"],"metadata":{"id":"7XusOvD8fZXh"},"execution_count":null,"outputs":[]},{"cell_type":"code","source":["populator = _metadata.MetadataPopulator.with_model_file(\"/content/drive/MyDrive/Colab Notebooks/final/converted_model.tflite\")\n","populator.load_metadata_buffer(metadata_buf)\n","populator.load_associated_files([\"/content/drive/MyDrive/Colab Notebooks/labels.txt\"])\n","populator.populate()"],"metadata":{"id":"yBAXRF1rfd4z"},"execution_count":null,"outputs":[]}]}