Simple line using only jetson util

j.perez · September 13, 2020, 11:13pm

In a simple program like this:

import jetson.inference
import jetson.utils

net = jetson.inference.detectNet(“ssd-mobilenet-v2”, threshold=0.5)
camera = jetson.utils.videoSource(“csi://0”) # ‘/dev/video0’ for V4L2
display = jetson.utils.videoOutput(“display://0”) # ‘my_video.mp4’ for file

while display.IsStreaming():
img = camera.Capture()
detections = net.Detect(img)
display.Render(img)
display.SetStatus(“Object Detection | Network {:.0f} FPS”.format(net.GetNetworkFPS()))

Are there any way to draw lines, rectangles and circles to the img frame, without using openCV?

AastaLLL · September 24, 2020, 2:49am

Hi,

Jetson_inference has a implementation to overlay input image with a filled rectangle here:

github.com

dusty-nv/jetson-utils/blob/master/cuda/cudaOverlay.cu#L89


		launch_overlay(uint8_t);

	else if( format == IMAGE_GRAY32F )

		launch_overlay(float);

	

	return cudaGetLastError();

}	

							 

							 

//----------------------------------------------------------------------------						 

template<typename T>

__global__ void gpuRectFill( T* input, T* output, int width, int height,

					    float4* rects, int numRects, float4 color ) 

{

	const int x = blockIdx.x * blockDim.x + threadIdx.x;

	const int y = blockIdx.y * blockDim.y + threadIdx.y;



	if( x >= width || y >= height )

		return;



	T px = input[ y * width + x ];

You can implement the overlay for lines and circles based on this.
Thanks.