feat: afdd make scripts

Author: CatchZeng

.vscode/settings.json

@@ -5,6 +5,7 @@
     "conda",
     "dtype",
     "imshow",
+    "ipynb",
     "pbtxt",
     "protoc",
     "pyplot",

Makefile

@@ -27,7 +27,7 @@ workspace-box:
 workspace-mask:
 	python scripts/workspace/mask/workspace.py --save_dir=$(SAVE_DIR) --name=$(NAME)
 	cp -r scripts/workspace/mask/files/* $(SAVE_DIR)/$(NAME)
-	python scripts/workspace/mask/replace.py --dir=$(SAVE_DIR)/$(NAME)/deployments --pattern="{{.Name}}" --repl=$(NAME)
+	python scripts/workspace/mask/replace.py --dir=$(SAVE_DIR)/$(NAME) --pattern="{{.Name}}" --repl=$(NAME)
 	cp models/research/object_detection/model_main_tf2.py $(SAVE_DIR)/$(NAME)
 	cp models/research/object_detection/exporter_main_v2.py $(SAVE_DIR)/$(NAME)
 	touch $(SAVE_DIR)/$(NAME)/annotations/label_map.pbtxt

scripts/workspace/mask/files/Makefile

@@ -32,4 +32,17 @@ export:
 	--input_type image_tensor \
 	--pipeline_config_path models/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8/pipeline.config \
 	--trained_checkpoint_dir models/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8/ \
-	--output_directory exported-models/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8
+	--output_directory exported-models/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8
+serving:
+	docker pull tensorflow/serving
+	docker run -d --name serving_base tensorflow/serving
+	# https://stackoverflow.com/a/45552938/14785930
+	docker exec -it serving_base mkdir -p /models/{{.Name}}/1
+	docker cp exported-models/mask_rcnn_inception_resnet_v2_1024x1024_coco17_gpu-8/saved_model/. serving_base:/models/{{.Name}}/1
+	docker commit --change "ENV MODEL_NAME {{.Name}}" serving_base {{.Name}}
+	docker rm -f serving_base
+serving-test:
+	python deployments/serving/client.py
+flask:
+	cp annotations/label_map.pbtxt deployments/flask
+	cd deployments/flask && make build

scripts/workspace/mask/replace.py

@@ -23,6 +23,9 @@ def replace(file_path, pattern, repl):
 def get_files(path, all_files):
     file_list = os.listdir(path)
     for file in file_list:
+        if ".ipynb" in file:
+            print('escape ' + file)
+            continue
         cur_path = os.path.join(path, file)
         if os.path.isdir(cur_path):
             get_files(cur_path, all_files)