Skip to content
Snippets Groups Projects
Commit 0fee1221 authored by jsteinmetz's avatar jsteinmetz
Browse files

Changed file paths and x and y sizes.

parent 79e435bd
No related branches found
No related tags found
No related merge requests found
import ml_module as T
xgb_model_path = 'TrainedModel/LightRedRaw_test_two.pkl'
xgb_model_path = "P:/work/ima/CNN/DarkRedJPG_test_no_opt_Caro.pkl"
# img_p = 'D:/TestImagesCrackDetection/In01/cropped_V13#01_0211.png'
# img_p = f'P:/work/ima/results/ProbeV12#Ref03/cropped_png/cropped_V12#Ref03_0500.png'
img_p = 'P:/work/ima/CNN/TrainingImagesManualMark/LightRedRaw/cropped_images/cropped_V13#02_3781.png'
img_p = f'P:/work/ima/results/ProbeV12#Ref03/cropped_png/cropped_V12#Ref03_0500.png'
# img_p = 'P:/work/ima/CNN/TrainingImagesManualMark/LightRedRaw/cropped_images/cropped_V13#02_3781.png'
SIZE_X = 327
SIZE_Y = 6307
SIZE_X = 256
SIZE_Y = 6096
DL_model = T.create_dl_model(size_x=SIZE_X, size_y=SIZE_Y)
XGB_model = T.load_xgb_model(xgb_model_path)
......@@ -17,6 +17,6 @@ XGB_model = T.load_xgb_model(xgb_model_path)
output_image, y, x = T.crack_image(image_path=img_p, size_x=SIZE_X, size_y=SIZE_Y, dl_model=DL_model,
xgb_model=XGB_model)
T.cv2.imwrite("C:/Users/jsteinmetz/Desktop/TestV13#02_3781.png", output_image)
T.cv2.imwrite("C:/Users/jsteinmetz/Desktop/TestV12#Ref03_0500_no_opt_Caro.png", output_image)
# print(T.crack_image(image_path=img_p, size_x=SIZE_X, size_y=SIZE_Y, dl_model=DL_model, xgb_model=XGB_model)[1])
# print(T.crack_image(image_path=img_p, size_x=SIZE_X, size_y=SIZE_Y, dl_model=DL_model, xgb_model=XGB_model)[2])
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment