YOLOV2 training using GPU

9 visualizaciones (últimos 30 días)
gil
gil el 29 de Nov. de 2021
Respondida: Shivam Singh el 3 de Dic. de 2021
Hi
Trying to run the following code on GPUCoder and error that trainYOLOv2ObjectDetector is not supported.
[detector,info] = trainYOLOv2ObjectDetector(trainingData,lgraph,options);
If the above code is not supported by the GPU coder, what are the alternatives to accelerate YOLOV2 training process?
Regards

Respuesta aceptada

Shivam Singh
Shivam Singh el 3 de Dic. de 2021
Hello gil,
Code generation using GPU Coder™ for trainYOLOv2ObjectDetector is not supported.
Yolov2 training process can be accelerated by using GPUs.
[detector,info] = trainYOLOv2ObjectDetector(trainingData,lgraph,options);
To access the GPUs, you can set the “options” in the trainYOLOv2ObjectDetector function, by referring the ExecutionEnvironment in the trainingOptions function.
Once the detector is trained and evaluated, then you can generate code for the yolov2ObjectDetector using GPU Coder™. For more information related to Code generation using GPU Coder™, you can refer the following:

Más respuestas (0)

Categorías

Más información sobre Deep Learning with GPU Coder en Help Center y File Exchange.

Etiquetas

Productos


Versión

R2021b

Community Treasure Hunt

Find the treasures in MATLAB Central and discover how the community can help you!

Start Hunting!

Translated by