// Prediction
// Accepts a single integral image/channel, thus only to be used for testing
// Assumes that predPtr is already allocated, of same size as imgPtr
DLL_EXPORT
void predictWithChannel( void *modelPtr, ImagePixelType *imgPtr,
IntegralImagePixelType *chImgPtr,
int width, int height, int depth,
PredictionPixelType *predPtr )
{
Matrix3D<PredictionPixelType> predMatrix;
predMatrix.fromSharedData( predPtr, width, height, depth );
// create roi for image, no GT available
ROIData roi;
roi.init( imgPtr, 0, 0, 0, width, height, depth );
// get the precomputed integral images
ROIData::IntegralImageType ii;
ii.fromSharedData(chImgPtr, width, height, depth);
roi.addII( ii.internalImage().data() );
MultipleROIData allROIs;
allROIs.add( shared_ptr_nodelete(ROIData, &roi) );
Booster adaboost;
adaboost.setModel( *((BoosterModel *) modelPtr) );
adaboost.predict<false>( allROIs, &predMatrix );
}
// Prediction for a single ROI
// Accepts an arbitrary number of integral images/channels.
// Assumes that predPtr is already allocated, of same size as imgPtr
// Returns 0 if ok
DLL_EXPORT
int predictWithChannels( void *modelPtr, ImagePixelType *imgPtr,
void *eigVecImgPtr,
int width, int height, int depth,
IntegralImagePixelType **chImgPtr,
int numChannels, double zAnisotropyFactor,
int useEarlyStopping,
int useROI, int x1, int y1, int z1, int x2, int y2, int z2, // used only if useROI != 0
PredictionPixelType *predPtr )
{
Matrix3D<PredictionPixelType> predMatrix;
predMatrix.fromSharedData( predPtr, width, height, depth );
// create roi for image, no GT available
ROIData roi;
roi.init( imgPtr, 0, 0, 0, width, height, depth, zAnisotropyFactor, 0.0, (const ROIData::RotationMatrixType *) eigVecImgPtr );
std::unique_ptr<ROIData::IntegralImageType[]> ii(new ROIData::IntegralImageType[numChannels]); // TODO: remove
for (int ch=0; ch < numChannels; ch++)
{
ii[ch].fromSharedData(chImgPtr[ch], width, height, depth);
roi.addII( ii[ch].internalImage().data() );
}
MultipleROIData allROIs;
allROIs.add( shared_ptr_nodelete(ROIData, &roi) );
try
{
Booster adaboost;
adaboost.setModel( *((BoosterModel *) modelPtr) );
if(useEarlyStopping != 0)
{
if (useROI)
{
ROICoordinates subROI;
subROI.x1 = x1; subROI.y1 = y1; subROI.z1 = z1;
subROI.x2 = x2; subROI.y2 = y2; subROI.z2 = z2;
adaboost.predictWithFeatureOrdering<true>( allROIs, &predMatrix, 0, IIBOOST_NUM_THREADS, &subROI );
}
else
adaboost.predictWithFeatureOrdering<true>( allROIs, &predMatrix );
}
else
adaboost.predictWithFeatureOrdering<false>( allROIs, &predMatrix );
}
catch( std::exception &e )
{
printf("Error in prediction: %s\n", e.what());
return -1;
}
return 0;
}
DLL_EXPORT
int predictIndividualWeakLearnersWithChannels( void *modelPtr, ImagePixelType *imgPtr,
void *eigVecImgPtr,
int width, int height, int depth,
IntegralImagePixelType **chImgPtr,
int numChannels, double zAnisotropyFactor,
WLPredictionPixelType **predPtr)
{
BoosterModel* model = static_cast<BoosterModel*>(modelPtr);
int numWL = model->size();
Matrix3D<WLPredictionPixelType> predMatrix[numWL]; // TODO: remove
for(int i=0; i < numWL; ++i)
predMatrix[i].fromSharedData(predPtr[i], width, height, depth);
// create roi for image, no GT available
ROIData roi;
roi.init( imgPtr, 0, 0, 0, width, height, depth, zAnisotropyFactor, 0.0, (const ROIData::RotationMatrixType *) eigVecImgPtr );
ROIData::IntegralImageType ii[numChannels]; // TODO: remove
for (int ch=0; ch < numChannels; ch++)
{
ii[ch].fromSharedData(chImgPtr[ch], width, height, depth);
roi.addII( ii[ch].internalImage().data() );
}
MultipleROIData allROIs;
allROIs.add( shared_ptr_nodelete(ROIData, &roi) );
try
{
Booster adaboost;
adaboost.setModel(*model);
adaboost.predictIndividualWeakLearners(allROIs, predMatrix);
}
catch( std::exception &e )
{
printf("Error in prediction: %s\n", e.what());
return -1;
}
return 0;
}
