Xception Check

Dockerfile

FROM tensorflow/tensorflow:latest-py3

RUN pip install pillow

main.go

import sys
import numpy as np
from PIL import Image
from tensorflow.python.keras.applications.xception import Xception, preprocess_input
from tensorflow.python.keras.models import Model

image = Image.open(sys.argv[1]).convert('RGB')
l = np.asarray(image.resize((299,299))).tolist()

base_model = Xception(include_top=True, weights='imagenet')
xception = Model(inputs=base_model.inputs, outputs=base_model.output)
raw_image = preprocess_input(np.expand_dims(l, axis=0))
vec = np.squeeze(xception.predict(raw_image)).tolist()

print("index, value")
for i, e in enumerate(vec):
    if e > 0.001953125:
        print(f"{i}, {e}") 

Makefile

build:
  docker build -t tensorflow-pil .
run:
  docker run -it --rm -v $(pwd):/workdir -w /workdir tensorflow-pil python main.py ${IMAGE_PATH}

ref for index: https://gist.github.com/xkumiyu/dd200f3f51986888c9151df4f2a9ef30