AntonMu
74 linhas
2.2 KiB
Python
74 linhas
2.2 KiB
Python
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))
|