Skip to content
Snippets Groups Projects
Commit 59f84a35 authored by Julian Steinmetz's avatar Julian Steinmetz
Browse files

Minor changes.

parent 106e72a2
No related branches found
No related tags found
No related merge requests found
import ml_module as T
xgb_model_path = 'TrainedModel/xgb_model.pkl'
xgb_model_path = 'TrainedModel/LightRedRaw_test_two.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 = 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 = 240
SIZE_Y = 6000
SIZE_X = 327
SIZE_Y = 6307
DL_model = T.create_dl_model(size_x=SIZE_X, size_y=SIZE_Y)
XGB_model = T.load_xgb_model(xgb_model_path)
......@@ -16,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/Test.png", output_image)
T.cv2.imwrite("C:/Users/jsteinmetz/Desktop/TestV13#02_3781.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