def __init__(self): directory = Paths.objects_dir ensurePathExists(directory) self.__directory = directory self.__objects = [] # A list of the loaded objects # Set different filters that can be used in self.getObjectIDList(objFilter=[One of the following]) self.TRACKABLE = Trackable self.TRACKABLEOBJ = TrackableObject self.TRACKABLEGROUP = TrackableGroupObject self.PICKUP = "PICKUP" # Any trackable object + group, NOT incuding the Robot Marker object self.MOTIONPATH = MotionPath
def __init__(self, directory): ensurePathExists(directory) self.__directory = directory self.__objects = [] # A list of the loaded objects # Set different filters that can be used in self.getObjectIDList(objFilter=[One of the following]) self.TRACKABLE = Resources.Trackable self.TRACKABLEOBJ = Resources.TrackableObject self.TRACKABLEGROUP = Resources.TrackableGroupObject self.MOTIONPATH = Resources.MotionPath self.RESOURCE = Resources.Resource self.FUNCTION = Resources.Function self.PICKUP = "PICKUP" # Any trackable object + group, NOT incuding the Robot Marker object self.__loadAllObjects()
def save(self, directory): """ Everything goes into a folder called TrackableObject OBJECTNAMEHERE Saves images for each View, with the format View#, and a data.txt folder is saved as a json, with this structure: { "Orientation_1": { "rect": (0, 0, 10, 10), "pickupRect": (0, 0, 10, 10), }, "Orientation_1": { "rect": (0, 0, 10, 10), "pickupRect": (0, 0, 10, 10), }, } """ # Make sure the "objects" directory exists ensurePathExists(directory) dataJson = {"tags": self.__tags, "Orientations": {}} # Save images and numpy arrays as seperate folders for index, view in enumerate(self.views): # Save the image filename = os.path.join(directory, "Orientation_" + str(index) + "_Image.png") cv2.imwrite(filename, view.image) # Add any view data to the dataJson dataJson["Orientations"]["Orientation_" + str(index)] = {"rect": view.rect, "pickupRect": view.pickupRect, "height": view.height} filename = os.path.join(directory, "data.txt") json.dump(dataJson, open(filename, 'w'), sort_keys=False, indent=3, separators=(',', ': '))
def save(self, directory): ensurePathExists(directory) # Long form (human readable: json.dump(self.dataJson, open(directory + "data.txt", 'w'), sort_keys=False, separators=(',', ': '))
def save(self, directory): ensurePathExists(directory) # Long form (human readable: filename = os.path.join(directory, "data.txt") json.dump(self.dataJson, open(filename, 'w'), sort_keys=False, separators=(',', ': '))