My-detection

yujy9014 · June 3, 2020, 7:19am

When I run my-detection.py, I want to print only the Class ID. What is there?

AastaLLL · June 3, 2020, 9:16am

Hi,

You can modified the console output from here:

dusty-nv/jetson-inference/blob/master/python/bindings/PyDetectNet.cpp#L141


      
          static PyObject* PyDetection_ToString( PyDetection_Object* self )
          {
          	// get center coord
          	float cx = 0.0f;
          	float cy = 0.0f;
          
          
	self->det.Center(&cx, &cy);
          
          
	// format string
          	char str[1024];
          
          
	sprintf(str, 
          		   "<detectNet.Detection object>\n"
          		   "   -- ClassID: %i\n"
          		   "   -- Confidence: %g\n"
          		   "   -- Left:    %g\n"
          		   "   -- Top:     %g\n"
          		   "   -- Right:   %g\n"
          		   "   -- Bottom:  %g\n"
          		   "   -- Width:   %g\n"
          		   "   -- Height:  %g\n"

Thanks.

dusty_nv · June 3, 2020, 5:21pm

You can also do this directly from the Python script using the detectNet.GetClassDesc() function, without modifying the C source:

for detection in detections:
    print(detection)
    print(net.GetClassDesc(detection.ClassID))