Vikas2425/Safety-Helmet-Detection

1import os
2import sys
3import pickle
4import numpy as np
5from src.logger import logging
6from src.exception import CustomException
7
8def save_object(file_path,obj):
9    try:
10        dir_path = os.path.dirname(file_path)
11        os.makedirs(dir_path, exist_ok=True)
12
13        with open(file_path, "wb") as file_obj:
14            pickle.dump(obj, file_obj)
15
16    except Exception as e:
17        raise CustomException(e, sys)
18    
19
20def load_object(file_path):
21    
22    try:
23        with open(file_path, "rb") as file_obj:
24            return pickle.load(file_obj)
25
26    except Exception as e:
27        raise CustomException(e, sys)