object_recognition.dbtools.add_db_arguments(parser) sched_group = parser.add_argument_group('Scheduler Options') from ecto.opts import scheduler_options scheduler_options(sched_group, default_scheduler='Threadpool') args = parser.parse_args() if args.compute_all == False and args.session_id == '': parser.print_usage() print "You must either supply a session id, or --all." sys.exit(1) return args if "__main__" == __name__: args = parse_args() couch = couchdb.Server(args.db_root) dbs = dbtools.init_object_databases(couch) sessions = dbs if args.compute_all: models.sync_models(dbs) results = models.Session.all(sessions) for session in results: generate_pointclouds_in_object_space(dbs, session, args) else: session = models.Session.load(sessions, args.session_id) if session == None or session.id == None: print "Could not load session with id:", args.session_id sys.exit(1) generate_pointclouds_in_object_space(dbs, session, args)
def parse_args(): parser = argparse.ArgumentParser(description='Copies docs from one db to another.') parser.add_argument('--remote', dest='remote', type=str, help='Remote db to copy from.') parser.add_argument('--remote_collection', dest='remote_collection', type=str, help='Remote collection.') object_recognition.dbtools.add_db_arguments(parser) args = parser.parse_args() return args if __name__ == "__main__": args = parse_args() couch = couchdb.Server(args.db_root) remote_couch = couchdb.Server(args.remote) local_db = dbtools.init_object_databases(couch) remote_db = remote_couch[args.remote_collection] results = remote_db.view('_all_docs') total_docs = len(results) i = 1 for x in results: doc = remote_db.get(x.id) doc = doc.copy() attachments = doc.get('_attachments',False) if attachments: del doc['_attachments'] if x.id not in local_db: (doc_id,rev) = local_db.save(doc) if attachments: for key,val in attachments.iteritems():
type=str, help='Remote db to copy from.') parser.add_argument('--remote_collection', dest='remote_collection', type=str, help='Remote collection.') object_recognition.dbtools.add_db_arguments(parser) args = parser.parse_args() return args if __name__ == "__main__": args = parse_args() couch = couchdb.Server(args.db_root) remote_couch = couchdb.Server(args.remote) local_db = dbtools.init_object_databases(couch) remote_db = remote_couch[args.remote_collection] results = remote_db.view('_all_docs') total_docs = len(results) i = 1 for x in results: doc = remote_db.get(x.id) doc = doc.copy() attachments = doc.get('_attachments', False) if attachments: del doc['_attachments'] if x.id not in local_db: (doc_id, rev) = local_db.save(doc) if attachments: for key, val in attachments.iteritems():
def parse_args(): import argparse parser = argparse.ArgumentParser( description='Train MyAlgorithm on views from the database.') parser.add_argument('objects', metavar='OBJECT', type=str, nargs='+', help='Object ids to train.') dbtools.add_db_arguments(parser) args = parser.parse_args() return args args = parse_args() db = dbtools.init_object_databases(couchdb.Server(args.db_root)) for object_id in args.objects: #get a list of observation ids for a particular object id. obs_ids = models.find_all_observations_for_object(db, object_id) if not obs_ids: print 'No observations found for object %s.' % object_id continue plasm = ecto.Plasm() #the db_reader transforms observation id into a set of image,depth,mask,K,R,T db_reader = capture.ObservationReader( "db_reader", db_params=dbtools.args_to_db_params(args)) #this iterates over all of the observation ids. observation_dealer = ecto.Dealer(
help='Turn on visiualization') object_recognition.dbtools.add_db_arguments(parser) sched_group = parser.add_argument_group('Scheduler Options') from ecto.opts import scheduler_options scheduler_options(sched_group, default_scheduler='Threadpool') args = parser.parse_args() if args.compute_all == False and args.session_id == '': parser.print_usage() print "You must either supply a session id, or --all." sys.exit(1) return args if "__main__" == __name__: args = parse_args() couch = couchdb.Server(args.db_root) dbs = dbtools.init_object_databases(couch) sessions = dbs if args.compute_all: models.sync_models(dbs) results = models.Session.all(sessions) for session in results: generate_pointclouds_in_object_space(dbs, session, args) else: session = models.Session.load(sessions, args.session_id) if session == None or session.id == None: print "Could not load session with id:", args.session_id sys.exit(1) generate_pointclouds_in_object_space(dbs, session, args)
from ecto_object_recognition import capture from object_recognition import models, tools as dbtools import ecto_opencv import os def parse_args(): import argparse parser = argparse.ArgumentParser(description='Train MyAlgorithm on views from the database.') parser.add_argument('objects', metavar='OBJECT', type=str, nargs='+', help='Object ids to train.') dbtools.add_db_arguments(parser) args = parser.parse_args() return args args = parse_args() db = dbtools.init_object_databases(couchdb.Server(args.db_root)) for object_id in args.objects: #get a list of observation ids for a particular object id. obs_ids = models.find_all_observations_for_object(db, object_id) if not obs_ids: print 'No observations found for object %s.' % object_id continue plasm = ecto.Plasm() #the db_reader transforms observation id into a set of image,depth,mask,K,R,T db_reader = capture.ObservationReader("db_reader", db_params=dbtools.args_to_db_params(args)) #this iterates over all of the observation ids. observation_dealer = ecto.Dealer(tendril=db_reader.inputs.at('observation'), iterable=obs_ids)