Python MLPVidの例

コード例 #1

ファイル: train_tfpv_kitti_vid_single_mlp_boot.py プロジェクト: slundqui/DeepGAP

    'learningRateBias': 1e-6,
    #'numClasses': trainDataObj.numClasses,
    'numClasses': trainDataObj.numClasses,
    'idxToName': trainDataObj.idxToName,
    'preTrain': False,
    'lossWeight': trainDataObj.lossWeight,
    'gtShape': trainDataObj.gtShape,
    'gtSparse': False,
    'inputScale': 100,
    'regWeight': 0,
    'resLoad': True,
}

#Allocate tensorflow object
#This will build the graph
tfObj = MLPVid(stage2_params, trainDataObj.inputShape)

print "Done init"
tfObj.runModel(trainDataObj, testDataObj = testDataObj)
print "Done run"

tfObj.closeSess()

stage3_params = {
    #Base output directory
    'outDir':          "/home/slundquist/mountData/DeepGAP/",
    #Inner run directory
    'runDir':          "/tfpv_kitti_vid_single_4x8_boot_3_bin/",
    'tfDir':           "/tfout",
    #Save parameters
    'ckptDir':         "/checkpoints/",

コード例 #2

ファイル: train_pv_kitti_vid_mlp_boot.py プロジェクト: slundqui/DeepGAP

    'learningRateBias': 1e-6,
    #'numClasses': trainDataObj.numClasses,
    'numClasses': trainDataObj.numClasses,
    'idxToName': trainDataObj.idxToName,
    'preTrain': False,
    'lossWeight': trainDataObj.lossWeight,
    'gtShape': trainDataObj.gtShape,
    'gtSparse': False,
    'inputScale': 100,
    'regWeight': 1e-4,
    'resLoad': True,
}

#Allocate tensorflow object
#This will build the graph
tfObj = MLPVid(stage2_params, trainDataObj.inputShape)

print "Done init"
tfObj.runModel(trainDataObj, testDataObj=testDataObj)
print "Done run"

tfObj.closeSess()

stage3_params = {
    #Base output directory
    'outDir': "/home/slundquist/mountData/DeepGAP/",
    #Inner run directory
    'runDir': "/pv_kitti_vid_4x8_boot_3_bin/",
    'tfDir': "/tfout",
    #Save parameters
    'ckptDir': "/checkpoints/",

コード例 #3

ファイル: eval_tfpv_kitti_vid_mlp_boot.py プロジェクト: slundqui/DeepGAP

        trainDataObj.lossWeight,
        'gtShape':
        trainDataObj.gtShape,
        'gtSparse':
        False,
        'inputScale':
        50,
        'regWeight':
        0,
        'resLoad':
        False,
    }

    #Allocate tensorflow object
    #This will build the graph
    tfObj = MLPVid(stage2_params, trainDataObj.inputShape)

    print "Done init"
    tfObj.evalModelBatch(testDataObj=testDataObj)
    print "Done run"

    tfObj.closeSess()

    testDataObj.imgIdx = 0
    stage3_params = {
        #Base output directory
        'outDir':
        "/home/slundquist/mountData/DeepGAP/",
        #Inner run directory
        'runDir':
        "/eval_tfpv_kitti_vid_4x8_boot_3_bin_" + runSuffix + "/",