Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
#test.py
#import dependencies
import cv2
import os
import MoCV
"""
_segment_image_test - function to test optimal thresholding for image segmentation
Parameters:
test_folder_path I/P path to test folder for image segmentation
test_img_name I/P name of test image
"""
def _segment_image_test(test_folder_path, img_name):
#read_image
img = cv2.imread(os.path.join(test_folder_path, img_name), 0)
#print image out
#cv2.imshow('original image', img)
#cv2.waitKey(0)
#cv2.destroyAllWindows()
#segment images front and back
front_img, back_img = MoCV.segmentation.optimal_thresholding(img)
front_img_path = os.path.join(test_folder_path, "front_segmented_img.png")
back_img_path = os.path.join(test_folder_path, "back_segmented_img.png")
#write front and back images
cv2.imwrite(front_img_path, front_img)
cv2.imwrite(back_img_path, back_img)
"""
test - function test all functions
Parameters:
mode I/P index of CV algorithsm for testing. Look at README for indices of algorithms. -1 for all algorithms
"""
_segment_image_test("path_to_image_folder, "name_of_image")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment