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 __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=(',', ': '))
