56 lines
1.6 KiB
Makefile
56 lines
1.6 KiB
Makefile
SHELL := /bin/bash
|
|
|
|
MAKEFILE_DIR := $(realpath $(dir $(lastword $(MAKEFILE_LIST))))
|
|
MEDIAPIPE_DIR := $(MAKEFILE_DIR)/../../..
|
|
|
|
BAZEL_COMPILATION_MODE ?= opt
|
|
BAZEL_TARGET ?= mediapipe/examples/coral:face_detection_tpu
|
|
BAZEL_CPU ?= k8
|
|
|
|
OUT_DIR := $(MEDIAPIPE_DIR)/out/$(BAZEL_CPU)
|
|
|
|
PLATFORM ?= amd64
|
|
DOCKER_FILE ?= $(MAKEFILE_DIR)/Dockerfile.$(PLATFORM)
|
|
DOCKER_COMMAND ?=
|
|
|
|
bazel_output = $(MEDIAPIPE_DIR)/bazel-bin/$(subst :,/,$(1))
|
|
|
|
define run_command
|
|
chmod a+w /; \
|
|
groupadd --gid $(shell id -g) $(shell id -g -n); \
|
|
useradd -m -e '' -s /bin/bash --gid $(shell id -g) --uid $(shell id -u) $(shell id -u -n); \
|
|
echo '$(shell id -u -n) ALL=(ALL) NOPASSWD:ALL' >> /etc/sudoers; \
|
|
su $(shell id -u -n) $(if $(1),-c '$(1)',)
|
|
endef
|
|
|
|
.PHONY: help
|
|
help:
|
|
@echo "make help - Print help"
|
|
@echo "make docker - Run Docker environment"
|
|
@echo "make build - Run Bazel build, use BAZEL_TARGET to choose which target to build"
|
|
|
|
ifeq (,$(wildcard /.dockerenv))
|
|
.PHONY: docker
|
|
docker:
|
|
docker run --rm -i --tty \
|
|
-v $(MEDIAPIPE_DIR):/mediapipe \
|
|
--workdir /mediapipe/ \
|
|
$(shell docker build -q - < $(DOCKER_FILE)) \
|
|
/bin/bash -c "$(call run_command,$(DOCKER_COMMAND))"
|
|
endif
|
|
|
|
.PHONY: build
|
|
build:
|
|
(cd $(MEDIAPIPE_DIR) && \
|
|
bazel build \
|
|
--crosstool_top=@crosstool//:toolchains \
|
|
--compiler=gcc \
|
|
--cpu=${BAZEL_CPU} \
|
|
--compilation_mode=${BAZEL_COMPILATION_MODE} \
|
|
--define darwinn_portable=1 \
|
|
--define MEDIAPIPE_DISABLE_GPU=1 \
|
|
--define MEDIAPIPE_EDGE_TPU=all \
|
|
$(BAZEL_TARGET) && \
|
|
mkdir -p $(OUT_DIR) && \
|
|
cp -f $(call bazel_output,$(BAZEL_TARGET)) $(OUT_DIR))
|