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)
