| import cv2 | |
| import os | |
| import os.path as osp | |
| from pathlib import Path | |
| import pickle | |
| def get_object(name): | |
| objects_dir = osp.join(Path(__file__).parent.absolute(), 'objects') | |
| if not name.endswith('.pkl'): | |
| name = name+".pkl" | |
| filepath = osp.join(objects_dir, name) | |
| if not osp.exists(filepath): | |
| return None | |
| with open(filepath, 'rb') as f: | |
| obj = pickle.load(f) | |
| return obj | |