代码拉取完成,页面将自动刷新
import os
import subprocess
import time
import sys
def make_call_string(arglist):
result_string = ""
for arg in arglist:
result_string += "".join(["--", arg[0], " ", arg[1], " "])
return result_string
root_folder = os.path.dirname(os.path.abspath(__file__))
data_folder = os.path.join(root_folder, "Data")
model_folder = os.path.join(data_folder, "Model_Weights")
image_folder = os.path.join(data_folder, "Source_Images")
input_folder = os.path.join(image_folder, "Test_Images")
output_folder = os.path.join(image_folder, "Test_Image_Detection_Results")
if not os.path.exists(output_folder):
os.mkdir(output_folder)
# First download the pre-trained weights
download_script = os.path.join(model_folder, "Download_Weights.py")
if not os.path.isfile(os.path.join(model_folder, "trained_weights_final.h5")):
print("\n", "Downloading Pretrained Weights", "\n")
start = time.time()
call_string = " ".join(
[
"python",
download_script,
"1MGXAP_XD_w4OExPP10UHsejWrMww8Tu7",
os.path.join(model_folder, "trained_weights_final.h5"),
]
)
subprocess.call(call_string, shell=True)
end = time.time()
print("Downloaded Pretrained Weights in {0:.1f} seconds".format(end - start), "\n")
# Now run the cat face detector
detector_script = os.path.join(
os.path.dirname(os.path.abspath(__file__)), "3_Inference", "Detector.py"
)
result_file = os.path.join(output_folder, "Detection_Results.csv")
model_weights = os.path.join(model_folder, "trained_weights_final.h5")
classes_file = os.path.join(model_folder, "data_classes.txt")
anchors = os.path.join(
root_folder, "2_Training", "src", "keras_yolo3", "model_data", "yolo_anchors.txt"
)
arglist = [
["input_path", input_folder],
["classes", classes_file],
["output", output_folder],
["yolo_model", model_weights],
["box_file", result_file],
["anchors", anchors],
["file_types", ".jpg .jpeg .png"],
]
call_string = " ".join(["python", detector_script, make_call_string(arglist)])
print("Detecting Cat Faces by calling: \n\n", call_string, "\n")
start = time.time()
subprocess.call(call_string, shell=True)
end = time.time()
print("Detected Cat Faces in {0:.1f} seconds".format(end - start))
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。