import _init_paths
from lib.data_utils import get_demo_h5, get_full_test
from lib.aligning import estimateSimilarityUmeyama
from global_info import global_info
def compose_rt(rotation, translation):
aligned_RT = np.zeros((4, 4), dtype=np.float32)
aligned_RT[:3, :3] = rotation.transpose()
aligned_RT[:3, 3] = translation
aligned_RT[3, 3] = 1
return aligned_RT
if __name__ == '__main__':
infos = global_info()
my_dir = infos.base_path
parser = argparse.ArgumentParser()
parser.add_argument('--domain',
default='unseen',
help='which sub test set to choose')
parser.add_argument('--nocs',
default='ANCSH',
help='which nocs type to use')
parser.add_argument('--item',
default='oven',
help='object category for benchmarking')
parser.add_argument('--save',
action='store_true',
help='save err to pickles')
import geneticist, faction, city, output, grid, area, global_info
from vector3 import *
#create a random number generator
class random_generator:
def __init__(self, minimum, maximum):
self.mi = minimum
self.ma = maximum
def __call__(self, *args):
return (random.random() * (self.ma-self.mi)) + self.mi
#create some information about the factions we're going to create
info = global_info.global_info()
info.num_factions = 4
info.faction_colors = [ vector3(255,0,0), vector3(0,0,255),
vector3(255,255,0), vector3(0,255,255) ]
info.faction_names = [ "orcs", "humans", "elves", "fairies" ]
grid.global_info = info
area.global_info = info
#initialize pygame
pygame.init()
screen = pygame.display.set_mode( (800,600), 0, 24 )
screen.fill( (0,0,0) )
#create information display structures
out = output.output( screen, 14 )
