docker

Author: PINTO0309

.dockerignore

@@ -0,0 +1 @@
+*

.github/workflows/python-publish.yml

@@ -29,3 +29,33 @@ jobs:
       with:
         user: __token__
         password: ${{ secrets.PYPI_API_TOKEN }}
+
+  docker-deploy:
+    runs-on: ubuntu-latest
+    needs: pypi-deploy
+    permissions:
+      contents: read
+      packages: write
+    steps:
+      - name: Git checkout
+        uses: actions/checkout@v3
+      - name: Enable buildx
+        uses: docker/setup-buildx-action@v2
+      - name: Log in to the Container registry
+        uses: docker/login-action@f054a8b539a109f9f41c372932f1ae047eff08c9
+        with:
+          registry: ${{ env.REGISTRY }}
+          username: ${{ github.actor }}
+          password: ${{ secrets.GITHUB_TOKEN }}
+      - name: Extract metadata (tags, labels) for Docker
+        id: meta
+        uses: docker/metadata-action@98669ae865ea3cffbcbaa878cf57c20bbf1c6c38
+        with:
+          images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
+      - name: Build and push Docker image
+        uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
+        with:
+          context: .
+          push: true
+          tags: ${{ steps.meta.outputs.tags }}
+          labels: ${{ steps.meta.outputs.labels }}

.gitignore

@@ -5,4 +5,5 @@ __pycache__/
 *.tflite
 *.json
 _version.py
+.bash_history
 schema.fbs

Dockerfile

@@ -0,0 +1,33 @@
+FROM ubuntu:22.04
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+SHELL ["/bin/bash", "-c"]
+
+RUN apt-get update \
+    && apt-get install -y \
+        python3-pip \
+        python-is-python3 \
+        flatbuffers-compiler \
+        sudo \
+    && pip install -U pip \
+    && pip install -U tfliteiorewriter \
+    && pip install -U tfliteiorewriter \
+    && apt clean \
+    && rm -rf /var/lib/apt/lists/*
+
+ENV USERNAME=user
+RUN echo "root:root" | chpasswd \
+    && useradd \
+        --create-home \
+        --home-dir /home/${USERNAME} \
+        --shell /bin/bash \
+        --user-group \
+        --groups adm,sudo \
+        ${USERNAME} \
+    && echo "${USERNAME}:${USERNAME}" | chpasswd \
+    && cat /dev/null > /etc/sudoers.d/${USERNAME} \
+    && echo "%${USERNAME}    ALL=(ALL)   NOPASSWD:    ALL" >> \
+        /etc/sudoers.d/${USERNAME}
+USER ${USERNAME}
+WORKDIR /home/${USERNAME}

README.md

@@ -20,13 +20,30 @@ The purpose is to solve the following problems by forcibly rewriting tflite's in
   ![02](https://github.com/PINTO0309/tflite-input-output-rewriter/assets/33194443/5d73d9e1-cae3-498f-8de6-371a8ddb9ce6)
 
 ## Execution
-```bash
-sudo apt-get update && sudo apt-get install -y flatbuffers-compiler
-
-python tfliteiorewriter/main.py -i xxxx.tflite
-```
+1. Docker
+    ```
+    $ docker login ghcr.io
+
+    Username (xxxx): {Enter}
+    Password: {Personal Access Token}
+    Login Succeeded
+
+    $ docker run --rm -it \
+    -v `pwd`:/home/user \
+    ghcr.io/pinto0309/tfliteiorewriter:latest
+
+    tfliteiorewriter -i xxxx.tflite
+    ```
+2. Local
+    ```bash
+    sudo apt-get update && sudo apt-get install -y flatbuffers-compiler
+    # Other than debian/ubuntu: https://github.com/google/flatbuffers/releases
+    pip install -U tfliteiorewriter
+
+    tfliteiorewriter -i xxxx.tflite
+    ```
 ```
-usage: main.py [-h] -i INPUT_TFLITE_FILE_PATH [-v] [-o OUTPUT_FOLDER_PATH] [-r RENAME RENAME]
+usage: tfliteiorewriter [-h] -i INPUT_TFLITE_FILE_PATH [-v] [-o OUTPUT_FOLDER_PATH] [-r RENAME RENAME]
 
 optional arguments:
   -h, --help
@@ -66,7 +83,7 @@ If this tool is run without additional options, it will overwrite the input/outp
 
 ## View Mode Result
 ```bash
-python tfliteiorewriter/main.py -i xxxx.tflite -v
+tfliteiorewriter -i xxxx.tflite -v
 ```
 ![image](https://github.com/PINTO0309/tflite-input-output-rewriter/assets/33194443/0d43d93d-647d-40f1-b464-662e39dcf228)
 
@@ -78,7 +95,7 @@ Replace with any name by specifying `{From}` and `{To}` in the `--renmae (-r)` o
   ![image](https://github.com/PINTO0309/tflite-input-output-rewriter/assets/33194443/61195485-a756-4449-8bf2-4d9e83f06feb)
 
 ```bash
-python tfliteiorewriter/main.py \
+tfliteiorewriter \
 -i xxxx.tflite \
 -r serving_default_input_1:0 aaa \
 -r StatefulPartitionedCall:0 bbb