HarleyHK / LARA-2023_Cpp-Fork-Detection

Rede neural YOLOv7-fork para detecção de garfos em C++ usando OpenCV e ONNXRunTime.

LARA-2023_Cpp-Fork-Detection

Este repositório contém o código para detecção de garfos com a rede neural YOLOv7-fork em C++ usando OpenCV e ONNXRunTime.

Configuração

• Ubuntu 20.04 - WSL 2
• Versão do gcc/g++: 11.1.0
• Versão do CMake: 3.16.3
• Versão do ONNXRunTime: 1.13.1
• Versão do OpenCV: 4.2.0

Crie o ambiente com CMake

cmake -S . -B build

Como usar

Compilar o código

(cd build/ && make)

Executar o código

fork_best_640x640.onnx fork_best_640x480.onnx

./main [MODEL] [INPUT] <optional>[OUTPUT]

MODEL: Caminho para o modelo;

INPUT: Caminho para a imagem de entrada; e

OUTPUT: Caminho para a imagem de saída (opcional).

• Exemplo

./main fork_best_640x640.onnx fork1.jpg fork1_pred.jpg

Resultados

Original	Predição