def test_demo2(): """Check that demo2 makes the same image using demo2.py and demo2.yaml. """ import demo2 print('Running demo2.py') demo2.main([]) logger = logging.getLogger('galsim') logger.setLevel(logging.WARNING) config = galsim.config.ReadConfig('demo2.yaml', file_type='yaml', logger=logger)[0] print('Running demo2.yaml') galsim.config.Process(config, logger=logger, except_abort=True) assert check_same('output/demo2.fits', 'output_yaml/demo2.fits')
def test_demo2(): """Check that demo2 makes the same image using demo2.py and demo2.yaml. """ import demo2 print("Running demo2.py") demo2.main([]) logging.basicConfig(format="%(message)s", level=logging.INFO, stream=sys.stdout) logger = logging.getLogger("galsim") config = galsim.config.ReadConfig("demo2.yaml", logger=logger)[0] print("Running demo2.yaml") galsim.config.Process(config, logger=logger) assert check_same("output/demo2.fits", "output_yaml/demo2.fits")
def test_demo2(): """Check that demo2 makes the same image using demo2.py and demo2.yaml. """ import demo2 print('Running demo2.py') demo2.main([]) logging.basicConfig(format="%(message)s", level=logging.INFO, stream=sys.stdout) logger = logging.getLogger('galsim') config = galsim.config.ReadConfig('demo2.yaml', logger=logger)[0] print('Running demo2.yaml') galsim.config.Process(config, logger=logger) assert check_same('output/demo2.fits', 'output_yaml/demo2.fits')
keypoints = keypoints.astype(np.float64) sent = 0 if len(keypoints.shape) > 0: rightHand = keypoints[ 0, 4] # right hand = [x horizontal value between 0 and 639, z vertical value between 0 and 479, confidence between 0 and 1] rightShoulder = keypoints[0, 2] rightElbow = keypoints[0, 3] if (rightHand[2] > 0.5) and (rightElbow[2] > 0.3) and (rightShoulder[2] > 0.3): # HMR model, joints3d, joints = demo2.main(color_image, keypoints, config2, model=model, vis=False) hmrRightWrist = joints3d[0, 6] hmrRightElbow = joints3d[0, 7] hmrRightShoulder = joints3d[0, 8] hmrLeftShoulder = joints3d[0, 9] hmrLeftElbow = joints3d[0, 10] hmrLeftWrist = joints3d[0, 11] if hmrRightShoulder[0] > hmrLeftShoulder[0]: print("backwards") else: armPixelLength3D = arm_length(hmrRightWrist, hmrRightElbow, hmrRightShoulder) armPixelLength = arm_length(rightHand[:-1], rightElbow[:-1],