utils.writeErrToScreen('renderErrs', renderErrs, epoch, j)
    utils.writeErrToScreen('renderGtErrs', renderGtErrs, epoch, j)

    utils.writeErrToFile('directErrs', directErrs, testingLog, epoch, j)
    utils.writeErrToFile('globalIllu2', globalIllu2Errs, testingLog, epoch, j)
    utils.writeErrToFile('globalIllu3', globalIllu3Errs, testingLog, epoch, j)
    utils.writeErrToFile('renderErrs', renderErrs, testingLog, epoch, j)
    utils.writeErrToFile('renderGtErrs', renderGtErrs, testingLog, epoch, j)

    directErrsNpList = np.concatenate( [directErrsNpList, utils.turnErrorIntoNumpy(directErrs) ], axis=0)
    globalIllu2ErrsNpList = np.concatenate( [globalIllu2ErrsNpList, utils.turnErrorIntoNumpy(globalIllu2Errs)], axis=0)
    globalIllu3ErrsNpList = np.concatenate( [globalIllu3ErrsNpList, utils.turnErrorIntoNumpy(globalIllu3Errs)], axis=0)
    renderErrsNpList = np.concatenate( [renderErrsNpList, utils.turnErrorIntoNumpy(renderErrs)], axis=0 )
    renderGtErrsNpList = np.concatenate( [renderGtErrsNpList, utils.turnErrorIntoNumpy(renderGtErrs)], axis=0 )

    utils.writeNpErrToScreen('globalIllu2_Accu:', np.mean(globalIllu2ErrsNpList[1:j+1, :], axis=0), epoch, j)
    utils.writeNpErrToScreen('globalIllu3_Accu:', np.mean(globalIllu3ErrsNpList[1:j+1, :], axis=0), epoch, j)
    utils.writeNpErrToScreen('directErrs_Accu:', np.mean(directErrsNpList[1:j+1, :], axis=0), epoch, j)
    utils.writeNpErrToScreen('renderErrs_Accu:', np.mean(renderErrsNpList[1:j+1, :], axis=0), epoch, j)
    utils.writeNpErrToScreen('renderGtErrs_Accu:', np.mean(renderGtErrsNpList[1:j+1, :], axis=0), epoch, j)

    utils.writeNpErrToFile('globalIllu2_Accu:', np.mean(globalIllu2ErrsNpList[1:j+1, :], axis=0), testingLog, epoch, j)
    utils.writeNpErrToFile('globalIllu3_Accu:', np.mean(globalIllu3ErrsNpList[1:j+1, :], axis=0), testingLog, epoch, j)
    utils.writeNpErrToFile('directErrs_Accu:', np.mean(directErrsNpList[1:j+1, :], axis=0), testingLog, epoch, j)
    utils.writeNpErrToFile('renderErrs_Accu:', np.mean(renderErrsNpList[1:j+1, :], axis=0), testingLog, epoch, j)
    utils.writeNpErrToFile('renderGtErrs_Accu:', np.mean(renderGtErrsNpList[1:j+1, :], axis=0), testingLog, epoch, j)


    if j == 1 or j % 2000 == 0:
        # Save the ground truth and the input
        vutils.save_image( (0.5*(albedoBatch + 1)*segBatch.expand_as(albedoBatch) ).data,
             utils.turnErrorIntoNumpy(pointErrs)], axis=0)
        normalErrsNpList = np.concatenate(
            [normalErrsNpList,
             utils.turnErrorIntoNumpy(normalErrs)], axis=0)
        meanAngleErrsNpList = np.concatenate(
            [meanAngleErrsNpList,
             utils.turnErrorIntoNumpy(meanAngleErrs)],
            axis=0)
        medianAngleErrsNpList = np.concatenate(
            [medianAngleErrsNpList,
             utils.turnErrorIntoNumpy(medianAngleErrs)],
            axis=0)

        if j < 1000:
            utils.writeNpErrToScreen(
                'pointAccu', np.mean(pointErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'normalAccu', np.mean(normalErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'meanAngleAccu',
                np.mean(meanAngleErrsNpList[1:j + 1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen(
                'medianAngleAccu',
                np.mean(medianAngleErrsNpList[1:j + 1, :], axis=0), epoch, j)

            utils.writeNpErrToFile(
                'pointAccu', np.mean(pointErrsNpList[1:j + 1, :], axis=0),
                trainingLog, epoch, j)
            utils.writeNpErrToFile(
Пример #3
0
            [normalErrsNpList,
             utils.turnErrorIntoNumpy(normalErrs)], axis=0)
        roughErrsNpList = np.concatenate(
            [roughErrsNpList,
             utils.turnErrorIntoNumpy(roughErrs)], axis=0)
        depthErrsNpList = np.concatenate(
            [depthErrsNpList,
             utils.turnErrorIntoNumpy(depthErrs)], axis=0)
        globalIllu1ErrsNpList = np.concatenate(
            [globalIllu1ErrsNpList,
             utils.turnErrorIntoNumpy(globalIllu1Errs)],
            axis=0)

        if j < 1000:
            utils.writeNpErrToScreen(
                'albedoAccu', np.mean(albedoErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'normalAccu', np.mean(normalErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'roughAccu', np.mean(roughErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'depthAccu', np.mean(depthErrsNpList[1:j + 1, :], axis=0),
                epoch, j)
            utils.writeNpErrToScreen(
                'globalIllu1Accu',
                np.mean(globalIllu1ErrsNpList[1:j + 1, :], axis=0), epoch, j)
            utils.writeNpErrToFile(
                'albedoAccu', np.mean(albedoErrsNpList[1:j + 1, :], axis=0),
        [globalIllu2ErrsNpList,
         utils.turnErrorIntoNumpy(globalIllu2Errs)],
        axis=0)
    globalIllu3ErrsNpList = np.concatenate(
        [globalIllu3ErrsNpList,
         utils.turnErrorIntoNumpy(globalIllu3Errs)],
        axis=0)
    imgEnvErrsNpList = np.concatenate(
        [imgEnvErrsNpList,
         utils.turnErrorIntoNumpy(imgEnvErrs)], axis=0)

    envErrsNpList = np.concatenate(
        [envErrsNpList, utils.turnErrorIntoNumpy(envErrs)], axis=0)

    utils.writeNpErrToScreen('albedoAccu',
                             np.mean(albedoErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen('normalAccu',
                             np.mean(normalErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen('roughAccu',
                             np.mean(roughErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen('depthAccu',
                             np.mean(depthErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)

    utils.writeNpErrToScreen(
        'globalIllu1Accu', np.mean(globalIllu1ErrsNpList[1:j + 1, :], axis=0),
        epoch, j)
    utils.writeNpErrToScreen(
Пример #5
0
        utils.writeErrToFile('globalIllu2', globalIllu2Errs, trainingLog,
                             epoch, j)
        utils.writeErrToFile('globalIllu3', globalIllu3Errs, trainingLog,
                             epoch, j)
        globalIllu2ErrsNpList = np.concatenate(
            [globalIllu2ErrsNpList,
             utils.turnErrorIntoNumpy(globalIllu2Errs)],
            axis=0)
        globalIllu3ErrsNpList = np.concatenate(
            [globalIllu3ErrsNpList,
             utils.turnErrorIntoNumpy(globalIllu3Errs)],
            axis=0)

        if j < 1000:
            utils.writeNpErrToScreen(
                'globalIllu2_Accu:',
                np.mean(globalIllu2ErrsNpList[1:j + 1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen(
                'globalIllu3_Accu',
                np.mean(globalIllu3ErrsNpList[1:j + 1, :], axis=0), epoch, j)
            utils.writeNpErrToFile(
                'globalIllu2_Accu',
                np.mean(globalIllu2ErrsNpList[1:j + 1, :], axis=0),
                trainingLog, epoch, j)
            utils.writeNpErrToFile(
                'globalIllu3_Accu',
                np.mean(globalIllu3ErrsNpList[1:j + 1, :], axis=0),
                trainingLog, epoch, j)
        else:
            utils.writeNpErrToScreen(
                'globalIllu2_Accu',
Пример #6
0
        utils.writeErrToScreen('normal', [normalErr], epoch, j )
        utils.writeErrToScreen('rough', [roughErr], epoch, j)
        utils.writeErrToScreen('depth', [depthErr], epoch, j)

        utils.writeErrToFile('albedo', [albedoErr], trainingLog, epoch, j)
        utils.writeErrToFile('normal', [normalErr], trainingLog, epoch, j)
        utils.writeErrToFile('rough', [roughErr], trainingLog, epoch, j)
        utils.writeErrToFile('depth', [depthErr], trainingLog, epoch, j)

        albedoErrsNpList = np.concatenate( [albedoErrsNpList, utils.turnErrorIntoNumpy( [albedoErr] )], axis=0)
        normalErrsNpList = np.concatenate( [normalErrsNpList, utils.turnErrorIntoNumpy( [normalErr] )], axis=0)
        roughErrsNpList = np.concatenate( [roughErrsNpList, utils.turnErrorIntoNumpy( [roughErr] )], axis=0)
        depthErrsNpList = np.concatenate( [depthErrsNpList, utils.turnErrorIntoNumpy( [depthErr] )], axis=0)

        if j < 1000:
            utils.writeNpErrToScreen('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), epoch, j )

            utils.writeNpErrToFile('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
        else:
            utils.writeNpErrToScreen('albedoAccu', np.mean(albedoErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('normalAccu', np.mean(normalErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('roughAccu', np.mean(roughErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('depthAccu', np.mean(depthErrsNpList[j-999:j+1, :], axis=0), epoch, j)

            utils.writeNpErrToFile('albedoAccu', np.mean(albedoErrsNpList[j-999:j+1, :], axis=0), trainingLog, epoch, j)
        [normal2ErrsNpList,
         utils.turnErrorIntoNumpy(normal2Errs)], axis=0)
    medianAngle2ErrsNpList = np.concatenate(
        [medianAngle2ErrsNpList,
         utils.turnErrorIntoNumpy(medianAngle2Errs)],
        axis=0)
    meanAngle2ErrsNpList = np.concatenate(
        [meanAngle2ErrsNpList,
         utils.turnErrorIntoNumpy(meanAngle2Errs)],
        axis=0)
    renderedErrsNpList = np.concatenate(
        [renderedErrsNpList,
         utils.turnErrorIntoNumpy(renderedErrs)], axis=0)

    utils.writeNpErrToScreen('normal1Accu',
                             np.mean(normal1ErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen(
        'medianAngle1Accu', np.mean(medianAngle1ErrsNpList[1:j + 1, :],
                                    axis=0), epoch, j)
    utils.writeNpErrToScreen('meanAngle1Accu',
                             np.mean(meanAngle1ErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen('normal2Accu',
                             np.mean(normal2ErrsNpList[1:j + 1, :], axis=0),
                             epoch, j)
    utils.writeNpErrToScreen(
        'medianAngle2Accu', np.mean(medianAngle2ErrsNpList[1:j + 1, :],
                                    axis=0), epoch, j)
    utils.writeNpErrToScreen('meanAngle2Accu',
                             np.mean(meanAngle2ErrsNpList[1:j + 1, :], axis=0),
Пример #8
0
        utils.writeErrToScreen('normal', [normalErr], epoch, j)
        utils.writeErrToScreen('rough', [roughErr], epoch, j)
        utils.writeErrToScreen('depth', [depthErr], epoch, j)

        utils.writeErrToFile('albedo', [albedoErr], trainingLog, epoch, j)
        utils.writeErrToFile('normal', [normalErr], trainingLog, epoch, j)
        utils.writeErrToFile('rough', [roughErr], trainingLog, epoch, j)
        utils.writeErrToFile('depth', [depthErr], trainingLog, epoch, j)

        albedoErrsNpList = np.concatenate( [albedoErrsNpList, utils.turnErrorIntoNumpy( [albedoErr] )], axis=0)
        normalErrsNpList = np.concatenate( [normalErrsNpList, utils.turnErrorIntoNumpy( [normalErr] )], axis=0)
        roughErrsNpList = np.concatenate( [roughErrsNpList, utils.turnErrorIntoNumpy( [roughErr] )], axis=0)
        depthErrsNpList = np.concatenate( [depthErrsNpList, utils.turnErrorIntoNumpy( [depthErr] )], axis=0)

        if j < 1000:
            utils.writeNpErrToScreen('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), epoch, j )

            utils.writeNpErrToFile('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
        else:
            utils.writeNpErrToScreen('albedoAccu', np.mean(albedoErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('normalAccu', np.mean(normalErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('roughAccu', np.mean(roughErrsNpList[j-999:j+1, :], axis=0), epoch, j)
            utils.writeNpErrToScreen('depthAccu', np.mean(depthErrsNpList[j-999:j+1, :], axis=0), epoch, j)

            utils.writeNpErrToFile('albedoAccu', np.mean(albedoErrsNpList[j-999:j+1, :], axis=0), trainingLog, epoch, j)
Пример #9
0
        utils.writeErrToFile('rough', [roughErr], trainingLog, epoch, j)
        utils.writeErrToFile('depth', [depthErr], trainingLog, epoch, j)

        utils.writeErrToFile('reconstErrors', [reconstErr], trainingLog, epoch, j)
        utils.writeErrToFile('renderErrors', [renderErr], trainingLog, epoch, j)

        albedoErrsNpList = np.concatenate( [albedoErrsNpList, utils.turnErrorIntoNumpy( [albedoErr] )], axis=0)
        normalErrsNpList = np.concatenate( [normalErrsNpList, utils.turnErrorIntoNumpy( [normalErr] )], axis=0)
        roughErrsNpList = np.concatenate( [roughErrsNpList, utils.turnErrorIntoNumpy( [roughErr] )], axis=0)
        depthErrsNpList = np.concatenate( [depthErrsNpList, utils.turnErrorIntoNumpy( [depthErr] )], axis=0)

        reconstErrsNpList = np.concatenate( [reconstErrsNpList, utils.turnErrorIntoNumpy( [reconstErr] )], axis=0 )
        renderErrsNpList = np.concatenate( [renderErrsNpList, utils.turnErrorIntoNumpy( [renderErr] )], axis=0 )

        if j < 1000:
            utils.writeNpErrToScreen('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), epoch, j )

            utils.writeNpErrToScreen('reconstAccu', np.mean(reconstErrsNpList[1:j+1, :], axis=0), epoch, j )
            utils.writeNpErrToScreen('renderAccu', np.mean(renderErrsNpList[1:j+1, :], axis=0), epoch, j )

            utils.writeNpErrToFile('albedoAccu', np.mean(albedoErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('normalAccu', np.mean(normalErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('roughAccu', np.mean(roughErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)
            utils.writeNpErrToFile('depthAccu', np.mean(depthErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j)

            utils.writeNpErrToFile('reconstAccu', np.mean(reconstErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j )
            utils.writeNpErrToFile('renderAccu', np.mean(renderErrsNpList[1:j+1, :], axis=0), trainingLog, epoch, j )
        else: