## @package db_input # Module caffe2.python.helpers.db_input def db_input(model, blobs_out, batch_size, db, db_type): dbreader_name = "dbreader_" + db dbreader = model.param_init_net.CreateDB( [], dbreader_name, db=db, db_type=db_type, ) return model.net.TensorProtosDBInput( dbreader, blobs_out, batch_size=batch_size)