mainC.py

# http://brm.io/matter-js/demo/#mixed
# http://www.pymunk.org/en/master/

from TestPrimitives import TestPrimitives
from TestPrimitives import TestRenderer
from TestPhysics import TestPhysics
from CapsuleNetwork import CapsuleNetwork
from HyperParameters import HyperParameters
from VividNet import VividNet
from Observation import Observation
from AttributePool import AttributePool
from Capsule import Capsule
from GraphicsUserInterface import GraphicsUserInterface
from numba import cuda, float32, int32
from Utility import Utility

import numpy as np
import random
import math

# We restrict our GPU to only use 50%. This has no specific 
# reason, just to allow us to work in the background without
# CUDA running out of Memory
import tensorflow as tf
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.5
set_session(tf.Session(config=config))

# The Primitive Capsules need to be more exact for this, thus we need (1 + 2) training loops
# and tighter HyperParameters. This is to avoid double detection.
HyperParameters.PrimitiveProbabilityCutOff = 0.80
HyperParameters.SemanticProbabilityCutOff  = 0.75

if __name__ == '__main__':
    testUI = GraphicsUserInterface()

    vividNet = VividNet("vividnetC")
    primCaps = vividNet.setRenderer(TestRenderer, TestPrimitives, 0) # , 2)

    # We train the Physics on Synthetic Data  
    vividNet.setSyntheticPhysics(TestPhysics, 0) 

    for i in range(10):
        vividNet._intuitivePhysics._physicsMemory._syntheticPhysics.generateInteractionSequence(vividNet._capsuleNetwork, 84, 84, "Tests/TSET/", str(i))

    exit()

    semCaps = vividNet.loadSemantic()
    print(semCaps)

    exit() 

    simObs, ignoreR = vividNet.showFrame("Examples/Bframe0.0.png")
    simObs, ignoreR = vividNet.showFrame("Examples/Bframe0.1.png") 

    # Print all Observations
    for capsule in simObs.keys():
        print(str(len(simObs[capsule])) + "x " + capsule.getName())
        for index, obs in enumerate(simObs[capsule]):
            print("Observation " + str(index))
            obs.printContinuity(False)
                    
    
    drawFrames = vividNet.renderPrediction(40)

    testUI.drawMovie(drawFrames, vividNet.getWidth(), vividNet.getHeight(), HyperParameters.TimeStep, False)