static void PostDisplay()
{
#ifndef _MSC_BUILD
// Write frames if necessary.
if (DumpFrames)
{
const int FRAME_INTERVAL = 1;
if ((FrameNumber % FRAME_INTERVAL) == 0)
{
const unsigned int w = glutGet(GLUT_WINDOW_WIDTH);
const unsigned int h = glutGet(GLUT_WINDOW_HEIGHT);
unsigned char * buffer = new char[w * h * 4];
glReadPixels(0, 0, w, h, GL_RGBA, GL_UNSIGNED_BYTE, buffer);
char filename[13];
sprintf(filename, "img%.5i.png", FrameNumber / FRAME_INTERVAL);
printf("Dumped %s.\n", filename);
saveImageRGBA(filename, buffer, w, h);
delete[] buffer;
}
FrameNumber++;
}
#endif
DrawFps();
glutSwapBuffers();
}
static void post_display ( void )
{
// Write frames if necessary.
if (dump_frames) {
const int FRAME_INTERVAL = 4;
if ((frame_number % FRAME_INTERVAL) == 0) {
const unsigned int w = glutGet(GLUT_WINDOW_WIDTH);
const unsigned int h = glutGet(GLUT_WINDOW_HEIGHT);
unsigned char * buffer = (unsigned char *) malloc(w * h * 4 * sizeof(unsigned char));
if (!buffer)
exit(-1);
// glRasterPos2i(0, 0);
glReadPixels(0, 0, w, h, GL_RGBA, GL_UNSIGNED_BYTE, buffer);
static char filename[80];
sprintf(filename, "snapshots/img%.5i.png", frame_number / FRAME_INTERVAL);
printf("Dumped %s.\n", filename);
saveImageRGBA(filename, buffer, w, h);
free(buffer);
}
}
frame_number++;
glutSwapBuffers ();
}
// main entry point
int main( int argc, char** argv )
{
printf("imagenet-console\n args (%i): ", argc);
for( int i=0; i < argc; i++ )
printf("%i [%s] ", i, argv[i]);
printf("\n\n");
// retrieve filename argument
if( argc < 2 )
{
printf("imagenet-console: input image filename required\n");
return 0;
}
const char* imgFilename = argv[1];
// create imageNet
imageNet* net = imageNet::Create(argc, argv);
if( !net )
{
printf("imagenet-console: failed to initialize imageNet\n");
return 0;
}
net->EnableProfiler();
// load image from file on disk
float* imgCPU = NULL;
float* imgCUDA = NULL;
int imgWidth = 0;
int imgHeight = 0;
if( !loadImageRGBA(imgFilename, (float4**)&imgCPU, (float4**)&imgCUDA, &imgWidth, &imgHeight) )
{
printf("failed to load image '%s'\n", imgFilename);
return 0;
}
float confidence = 0.0f;
// classify image
const int img_class = net->Classify(imgCUDA, imgWidth, imgHeight, &confidence);
if( img_class >= 0 )
{
printf("imagenet-console: '%s' -> %2.5f%% class #%i (%s)\n", imgFilename, confidence * 100.0f, img_class, net->GetClassDesc(img_class));
if( argc > 2 )
{
const char* outputFilename = argv[2];
// overlay the classification on the image
cudaFont* font = cudaFont::Create();
if( font != NULL )
{
char str[512];
sprintf(str, "%2.3f%% %s", confidence * 100.0f, net->GetClassDesc(img_class));
const int overlay_x = 10;
const int overlay_y = 10;
const int px_offset = overlay_y * imgWidth * 4 + overlay_x * 4;
// if the image has a white background, use black text (otherwise, white)
const float white_cutoff = 225.0f;
bool white_background = false;
if( imgCPU[px_offset] > white_cutoff && imgCPU[px_offset + 1] > white_cutoff && imgCPU[px_offset + 2] > white_cutoff )
white_background = true;
// overlay the text on the image
font->RenderOverlay((float4*)imgCUDA, (float4*)imgCUDA, imgWidth, imgHeight, (const char*)str, 10, 10,
white_background ? make_float4(0.0f, 0.0f, 0.0f, 255.0f) : make_float4(255.0f, 255.0f, 255.0f, 255.0f));
}
printf("imagenet-console: attempting to save output image to '%s'\n", outputFilename);
if( !saveImageRGBA(outputFilename, (float4*)imgCPU, imgWidth, imgHeight) )
printf("imagenet-console: failed to save output image to '%s'\n", outputFilename);
else
printf("imagenet-console: completed saving '%s'\n", outputFilename);
}
}
else
printf("imagenet-console: failed to classify '%s' (result=%i)\n", imgFilename, img_class);
printf("\nshutting down...\n");
CUDA(cudaFreeHost(imgCPU));
delete net;
return 0;
}
