aconvert.cpp

微软的基于HMM的人脸识别原代码, 非常经典的说

        src_img[i]->roi = &src_roi;
    }

    src_roi.coi = dst_roi.coi = 0;
    src_roi.xOffset = src_roi.yOffset = 
    dst_roi.xOffset = dst_roi.yOffset = 0;

    for( h = min_img_size; h <= max_img_size; )
    {
        for( w = min_img_size; w <= max_img_size; )
        {
            int      denom = (w - min_img_size + 1)*(h - min_img_size + 1)*channels;
            int      iters = (base_iters*2 + denom)/(2*denom);

            src_roi.width = dst_roi.width = w;
            src_roi.height = dst_roi.height = h;

            if( iters < 1 ) iters = 1;

            for( i = 0; i < iters; i++ )
            {
                double err0 = 0;

                switch( depth )
                {
                case IPL_DEPTH_8U:
                    atsRandSetBounds( &rng_state, 0, img8u_range );
                    break;
                case IPL_DEPTH_32F:
                    atsRandSetBounds( &rng_state, -img32f_range, img32f_range );
                    atsRandSetFloatBits( &rng_state, img32f_bits );
                    break;
                }

                src_roi.coi = 0;
                for( c = 1; c <= channels; c++ )
                {
                    atsFillRandomImageEx( src_img[c-1], &rng_state );
                }

                /* convert to pix. via IPL (single plane mode) */
                for( c = 1; c <= channels; c++ )
                {
                    dst_roi.coi = c;
                    iplCopy( src_img[c-1], dst_img );
                }

                dst_roi.coi = 0;
                cvCvtPlaneToPix( src_img[0], src_img[1], src_img[2],
                                 channels == 4 ? src_img[3] : 0, dst2_img );

                err0 = iplNorm( dst_img, dst2_img, IPL_C );

                if( err0 > max_err )
                {
                    merr_w    = w;
                    merr_h    = h;
                    merr_iter = i;
                    max_err   = err0;
                    if( max_err > success_error_level ) goto test_exit;
                }
            }
            ATS_INCREASE( w, img_size_delta_type, img_size_delta );
        } /* end of the loop by w */

        ATS_INCREASE( h, img_size_delta_type, img_size_delta );
    }  /* end of the loop by h */

test_exit:

    for( i = 0; i < channels; i++ )
    {
        src_img[i]->roi = 0;
        atsReleaseImage( src_img[i] );
    }
    
    dst_img->roi = dst2_img->roi = 0;
    atsReleaseImage( dst_img );
    atsReleaseImage( dst2_img );

    //if( code == TRS_OK )
    {
        trsWrite( ATS_LST, "Max err is %g at w = %d, h = %d, "
                           "iter = %d, seed = %08x",
                           max_err, merr_w, merr_h, merr_iter, seed );

        return max_err <= success_error_level ?
            trsResult( TRS_OK, "No errors" ) :
            trsResult( TRS_FAIL, "Bad accuracy" );
    }
    /*else
    {
        trsWrite( ATS_LST, "Fatal error at w = %d, h = %d, "
                           "iter = %d, seed = %08x",
                           w, h, i, seed );
        return trsResult( TRS_FAIL, "Function returns error code" );
    }*/
}


/* ///////////////////// cvt_scale_test ///////////////////////// */

static int cvt_scale_test( void* arg )
{
    static const int depths[] = { IPL_DEPTH_8U, IPL_DEPTH_8S, IPL_DEPTH_16S,
                                  IPL_DEPTH_32S, IPL_DEPTH_32F };
    
    double success_error_level = 1;

    int   param     = (int)arg;
    int   src_depth = depths[param / 5];
    int   dst_depth = depths[param % 5];
    int   seed     = atsGetSeed();

    /* position where the maximum error occured */
    int   merr_w = 0, merr_h = 0, merr_iter = 0;

    /* test parameters */
    int     w = 0, h = 0, i = 0, c = 0;
    double  err = 0, max_err = 0.;
    int     code = TRS_OK;

    IplImage    *dst_img, *dst2_img, *dst3_img, *src_img;
    AtsRandState rng_state;

    atsRandInit( &rng_state, 0, 1, seed );

    read_cvt_params();

    CvSize img_size = cvSize( max_img_size, max_img_size );

    src_img  = cvCreateImage( img_size, src_depth, 1 );
    dst_img  = cvCreateImage( img_size, dst_depth, 1 );
    dst2_img = cvCreateImage( img_size, IPL_DEPTH_32F, 1 );
    dst3_img = cvCreateImage( img_size, IPL_DEPTH_32F, 1 );

    if( dst_depth == IPL_DEPTH_32F )
    {
        success_error_level = 1e-5;
    }

    for( h = min_img_size; h <= max_img_size; )
    {
        for( w = min_img_size; w <= max_img_size; )
        {
            int  denom = (w - min_img_size + 1)*(h - min_img_size + 1);
            int  iters = (base_iters*2 + denom)/(2*denom);
            CvRect roi = cvRect(0,0,w,h);

            cvSetImageROI( src_img, roi );
            cvSetImageROI( dst_img, roi );
            cvSetImageROI( dst2_img, roi );
            cvSetImageROI( dst3_img, roi );

            if( iters < 1 ) iters = 1;

            for( i = 0; i < iters; i++ )
            {
                double err0 = 0;

                switch( src_depth )
                {
                case IPL_DEPTH_8U:
                    atsRandSetBounds( &rng_state, 0, img8u_range );
                    break;
                case IPL_DEPTH_8S:
                    atsRandSetBounds( &rng_state, -img8s_range, img8s_range );
                    break;
                case IPL_DEPTH_16S:
                    atsRandSetBounds( &rng_state, -img16s_range, img16s_range );
                    break;
                case IPL_DEPTH_32S:
                    atsRandSetBounds( &rng_state, -img16s_range*2, img16s_range*2 );
                    break;
                }

                atsFillRandomImageEx( src_img, &rng_state );

                for( c = 0; c < 2; c++ )
                {
                    double scale = 1, shift = 0;
                    
                    if( c == 1 )
                    {
                        scale = atsRandPlane32f( &rng_state )*4 - 2;
                        shift = atsRandPlane32f( &rng_state )*200 - 100;
                    }

                    if( scale == 0 ) scale = 0.001;

                    atsScaleAddImage( src_img, dst2_img, scale, shift );
                    if( dst_depth != IPL_DEPTH_32F )
                    {
                        atsConvert( dst2_img, dst_img );
                        atsConvert( dst_img, dst2_img );
                    }

                    cvConvertScale( src_img, dst_img, scale, shift );
                    atsConvert( dst_img, dst3_img );

                    err0 = iplNorm( dst2_img, dst3_img, IPL_C );

                    if( err > max_err )
                    {
                        merr_w    = w;
                        merr_h    = h;
                        merr_iter = i;
                        max_err   = err;
                        if( max_err > success_error_level )
                            goto test_exit;
                    }
                }
            }
            ATS_INCREASE( w, img_size_delta_type, img_size_delta );
        } /* end of the loop by w */

        ATS_INCREASE( h, img_size_delta_type, img_size_delta );
    }  /* end of the loop by h */

    code = TRS_OK;

test_exit:

    cvReleaseImage( &dst_img );
    cvReleaseImage( &dst2_img );
    cvReleaseImage( &dst3_img );
    cvReleaseImage( &src_img );

    trsWrite( ATS_LST, "Max err is %g at w = %d, h = %d, "
                       "iter = %d, seed = %08x",
                       max_err, merr_w, merr_h, merr_iter, seed );

    return max_err <= success_error_level ?
        trsResult( TRS_OK, "No errors" ) :
        trsResult( TRS_FAIL, "Bad accuracy" );
}


#define CVT_PIX2PLANE_8UC3     0
#define CVT_PIX2PLANE_8UAC4    1
#define CVT_PIX2PLANE_32FC3    2
#define CVT_PIX2PLANE_32FAC4   3

#define CVT_PLANE2PIX_8UC3     0
#define CVT_PLANE2PIX_8UAC4    1
#define CVT_PLANE2PIX_32FC3    2
#define CVT_PLANE2PIX_32FAC4   3


#define CVT_8U16S      2
#define CVT_8U32F      4
#define CVT_8S16S      7
#define CVT_8S32F      9
#define CVT_16S8U      10
#define CVT_16S8S      11
#define CVT_16S32F     14
#define CVT_32S32F     19
#define CVT_32F8U      20
#define CVT_32F8S      21
#define CVT_32F16S     22
#define CVT_32F32S     23

void InitAConvert( void )
{
    /* Registering test functions */
    trsRegArg( funcs[0], test_desc, atsAlgoClass, cvt_pix_to_plane_test, CVT_PIX2PLANE_8UC3 );
    trsRegArg( funcs[0], test_desc, atsAlgoClass, cvt_pix_to_plane_test, CVT_PIX2PLANE_8UAC4 );
    trsRegArg( funcs[0], test_desc, atsAlgoClass, cvt_pix_to_plane_test, CVT_PIX2PLANE_32FC3 );
    trsRegArg( funcs[0], test_desc, atsAlgoClass, cvt_pix_to_plane_test, CVT_PIX2PLANE_32FAC4 );

    trsRegArg( funcs[1], test_desc, atsAlgoClass, cvt_plane_to_pix_test, CVT_PLANE2PIX_8UC3 );
    trsRegArg( funcs[1], test_desc, atsAlgoClass, cvt_plane_to_pix_test, CVT_PLANE2PIX_8UAC4 );
    trsRegArg( funcs[1], test_desc, atsAlgoClass, cvt_plane_to_pix_test, CVT_PLANE2PIX_32FC3 );
    trsRegArg( funcs[1], test_desc, atsAlgoClass, cvt_plane_to_pix_test, CVT_PLANE2PIX_32FAC4 );

    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_8U32F );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_32F8U );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_8S32F );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_32F8S );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_16S32F );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_32F16S );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_32S32F );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_32F32S );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_8U16S );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_16S8U );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_8S16S );
    trsRegArg( funcs[2], test_desc, atsAlgoClass, cvt_scale_test, CVT_16S8S );

} /* InitAConvert */


/* End of file. */