get_masks.py File Reference

Namespaces
namespace	get_masks

Functions
	get_masks.extract_colored_class_masks (results, output_dir=None, image_name=None, class_colors=None)

Variables
str	get_masks.model_path = "/home/seame/ObjectDetectionAvoidance/models/yolo-object-lane-unfroze/weights/last.pt"
	get_masks.model = YOLO(model_path)
str	get_masks.folder_path = "/home/seame/seame_road"
str	get_masks.output_dir = "/home/seame/ObjectDetectionAvoidance/masks_last_conf"
dict	get_masks.custom_colors
	get_masks.file_path = os.path.join(folder_path, file_name)
	get_masks.results = model.predict(file_path, conf=0.1, imgsz=320)
	get_masks.image_name = os.path.splitext(os.path.basename(file_path))[0]
	get_masks.colored_mask = extract_colored_class_masks(results, output_dir, image_name, class_colors=custom_colors)