上传文件至 /
This commit is contained in:
parent
b44d0b4be7
commit
4317aa00da
156
CMakeLists.txt
Normal file
156
CMakeLists.txt
Normal file
@ -0,0 +1,156 @@
|
|||||||
|
cmake_minimum_required(VERSION 3.8)
|
||||||
|
|
||||||
|
project(rknn_yolo_demo_linux)
|
||||||
|
set(CMAKE_CXX_STANDARD 11)
|
||||||
|
add_definitions(-g -O0 -ggdb -gdwarf -funwind-tables -rdynamic)
|
||||||
|
add_definitions(-Wno-write-strings -Wno-return-type)
|
||||||
|
|
||||||
|
|
||||||
|
set(TOOLCHAIN_DIR /opt/atk-dlrv1126-toolchain)
|
||||||
|
set(CMAKE_CXX_COMPILER ${TOOLCHAIN_DIR}/usr/bin/arm-linux-gnueabihf-g++)
|
||||||
|
set(CMAKE_C_COMPILER ${TOOLCHAIN_DIR}/usr/bin/arm-linux-gnueabihf-gcc)
|
||||||
|
set(SYSROOT ${TOOLCHAIN_DIR}/arm-buildroot-linux-gnueabihf/sysroot/usr/include)
|
||||||
|
set(CMAKE_SYSROOT ${TOOLCHAIN_DIR}/arm-buildroot-linux-gnueabihf/sysroot)
|
||||||
|
|
||||||
|
set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS}")
|
||||||
|
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -pthread")
|
||||||
|
|
||||||
|
# for linux
|
||||||
|
if (CMAKE_C_COMPILER MATCHES "aarch64")
|
||||||
|
set(LIB_ARCH lib64)
|
||||||
|
else()
|
||||||
|
set(LIB_ARCH lib)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
# # for android
|
||||||
|
# if (${TARGET_SOC} STREQUAL "RK3399PRO")
|
||||||
|
# set(LIB_ARCH lib64)
|
||||||
|
# endif()
|
||||||
|
|
||||||
|
# rga
|
||||||
|
set(RGA_DIR ${MZ_ROOT}/libs/rklibs/rknpu-1.7.3/rknn/rknn_api/examples/3rdparty/rga)
|
||||||
|
include_directories(${RGA_DIR}/include)
|
||||||
|
|
||||||
|
# drm
|
||||||
|
set(DRM_DIR ${MZ_ROOT}/libs/common/drm)
|
||||||
|
include_directories(${DRM_DIR}/include)
|
||||||
|
include_directories(${DRM_DIR}/include/libdrm)
|
||||||
|
|
||||||
|
#stb
|
||||||
|
set(STB_DIR ${MZ_ROOT}/libs/common/)
|
||||||
|
include_directories(${STB_DIR})
|
||||||
|
|
||||||
|
# rknn api
|
||||||
|
if (${TARGET_SOC} STREQUAL "RK3399PRO")
|
||||||
|
set(RKNN_API_PATH ${MZ_ROOT}/libs/rklibs/RK3399Pro_npu/rknn-api/librknn_api)
|
||||||
|
include_directories(${RKNN_API_PATH}/include)
|
||||||
|
set(RKNN_API_LIB ${RKNN_API_PATH}/${CMAKE_SYSTEM_NAME}/${LIB_ARCH}/librknn_api.so)
|
||||||
|
else()
|
||||||
|
set(RKNN_API_PATH ${MZ_ROOT}/libs/rklibs/rknpu-1.7.3/rknn/rknn_api/librknn_api)
|
||||||
|
include_directories(${RKNN_API_PATH}/include)
|
||||||
|
set(RKNN_API_LIB ${RKNN_API_PATH}/${LIB_ARCH}/librknn_api.so)
|
||||||
|
endif()
|
||||||
|
|
||||||
|
# libjpeg
|
||||||
|
set(LIBJPEG_DIR ${MZ_ROOT}/libs/platform/${TARGET_SOC}/libjpeg)
|
||||||
|
include_directories(${LIBJPEG_DIR}/include)
|
||||||
|
set(LIBJPEG_LIB ${LIBJPEG_DIR}/lib/${CMAKE_SYSTEM_NAME}/libjpeg.a)
|
||||||
|
|
||||||
|
# libpng
|
||||||
|
set(LIBPNG_DIR ${MZ_ROOT}/libs/platform/${TARGET_SOC}/libpng)
|
||||||
|
include_directories(${LIBPNG_DIR}/include)
|
||||||
|
set(LIBPNG_LIB ${LIBPNG_DIR}/lib/${CMAKE_SYSTEM_NAME}/libpng.a)
|
||||||
|
|
||||||
|
# zlib(used by libpng)
|
||||||
|
set(ZLIB_DIR ${MZ_ROOT}/libs/platform/${TARGET_SOC}/zlib)
|
||||||
|
include_directories(${ZLIB_DIR}/include)
|
||||||
|
set(ZLIB_LIB ${ZLIB_DIR}/lib/${CMAKE_SYSTEM_NAME}/libz.a)
|
||||||
|
|
||||||
|
# #curl
|
||||||
|
include_directories(${CMAKE_SOURCE_DIR}/include)
|
||||||
|
# set(CURL_LIB /opt/atk-dlrv1126-toolchain/arm-buildroot-linux-gnueabihf/sysroot/usr/lib/libcurl.so)
|
||||||
|
|
||||||
|
set(libs
|
||||||
|
jes_bas
|
||||||
|
jes_upgrade
|
||||||
|
jes_user
|
||||||
|
jes_network
|
||||||
|
jes_mss
|
||||||
|
jes_isp
|
||||||
|
jes_mcs
|
||||||
|
jes_spk
|
||||||
|
jes_record
|
||||||
|
jes_recsnap
|
||||||
|
jes_replay
|
||||||
|
jes_storage
|
||||||
|
jlinux
|
||||||
|
jbus
|
||||||
|
jbuffer
|
||||||
|
dbus-1
|
||||||
|
jvbase
|
||||||
|
arch
|
||||||
|
utility
|
||||||
|
cjson
|
||||||
|
${LIBINTL})
|
||||||
|
|
||||||
|
link_directories(/opt/atk-dlrv1126-toolchain/arm-buildroot-linux-gnueabihf/sysroot/usr/lib)
|
||||||
|
find_package(CURL REQUIRED)
|
||||||
|
include_directories(${CURL_INCLUDE_DIR})
|
||||||
|
message("CURL_LIBRARIES: ${CURL_LIBRARIES}")
|
||||||
|
message("CURL_INCLUDE_DIR: ${CURL_INCLUDE_DIR}")
|
||||||
|
find_package(OpenCV REQUIRED)
|
||||||
|
find_package(OpenSSL REQUIRED)
|
||||||
|
|
||||||
|
set(OPENCV_LIBS opencv_core opencv_imgcodecs opencv_imgproc opencv_features2d opencv_flann opencv_highgui opencv_freetype)
|
||||||
|
set(RKNN_LIBS easymedia rga rkaiq sample_common_isp)
|
||||||
|
|
||||||
|
link_directories(include lib)
|
||||||
|
link_directories(include ./include/librtsp)
|
||||||
|
link_directories(/home/developer/sysroot/usr/lib)
|
||||||
|
link_directories(/home/developer/sysroot/lib)
|
||||||
|
|
||||||
|
include_directories(/home/developer/sysroot/usr/include/)
|
||||||
|
include_directories(${SYSROOT})
|
||||||
|
include_directories(${SYSROOT}/rga)
|
||||||
|
include_directories(${SYSROOT}/easymedia)
|
||||||
|
include_directories(${SYSROOT}/rkaiq/uAPI)
|
||||||
|
include_directories(${SYSROOT}/rkaiq/xcore)
|
||||||
|
include_directories(${SYSROOT}/rkaiq/algos)
|
||||||
|
include_directories(${SYSROOT}/rkaiq/common)
|
||||||
|
include_directories(${SYSROOT}/rkaiq/iq_parser)
|
||||||
|
include_directories(./include/librtsp)
|
||||||
|
include_directories(./include/nlohmann)
|
||||||
|
|
||||||
|
|
||||||
|
set(CMAKE_INSTALL_RPATH "lib")
|
||||||
|
|
||||||
|
add_executable(yolov5-JES
|
||||||
|
src/drm_func.c
|
||||||
|
src/rga_func.c
|
||||||
|
src/main.cpp
|
||||||
|
src/yolov5_detect_postprocess.cpp
|
||||||
|
src/yolov5_detect.cpp
|
||||||
|
)
|
||||||
|
|
||||||
|
target_link_libraries(yolov5-JES
|
||||||
|
${RKNN_API_LIB}
|
||||||
|
${LIBJPEG_LIB}
|
||||||
|
${LIBPNG_LIB}
|
||||||
|
${ZLIB_LIB}
|
||||||
|
${OPENCV_LIBS}
|
||||||
|
${CURL_LIBRARIES}
|
||||||
|
librtsp.a
|
||||||
|
${RKNN_LIBS}
|
||||||
|
dl
|
||||||
|
${libs}
|
||||||
|
OpenSSL::Crypto
|
||||||
|
OpenSSL::SSL
|
||||||
|
)
|
||||||
|
add_definitions(-DRKAIQ)
|
||||||
|
# install target and libraries
|
||||||
|
set(CMAKE_INSTALL_PREFIX ${CMAKE_SOURCE_DIR}/install/yolov5-JES)
|
||||||
|
install(TARGETS yolov5-JES DESTINATION ./)
|
||||||
|
install(PROGRAMS ${RKNN_API_LIB} DESTINATION lib )
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
64
README.md
Normal file
64
README.md
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
# rknn_yolo_EAI_pic
|
||||||
|
#### 简要说明
|
||||||
|
|
||||||
|
用于跑EAI提供仓库训练出来的yolov5模型的视频检测例程。
|
||||||
|
集成了从串口读取红外温度数据，到达阈值后报警并运行rknn模型
|
||||||
|
|
||||||
|
需注意运行该例程需要将npu驱动更新为1.7.3版本
|
||||||
|
|
||||||
|
本项目只适用于EAI-YOLOV5
|
||||||
|
详情请参考EAI官网：https://www.easy-eai.com/document_details/3/342
|
||||||
|
|
||||||
|
|
||||||
|
## 准备工作
|
||||||
|
根据EAI给出的教程训练、转化得到rknn模型
|
||||||
|
有两类检测模型：火焰检测和烟雾检测
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## 安装(install)
|
||||||
|
|
||||||
|
通过adb连接设备并将编译结果推送至板端，执行以下命令推送，这里板端默认使用/userdata路径
|
||||||
|
|
||||||
|
```
|
||||||
|
adb push install/rknn_yolo_demo /userdata/
|
||||||
|
```
|
||||||
|
|
||||||
|
将前面准备好的RKNN模型推送至板端，这里假设模型名字为 yolov5s_u8.rknn
|
||||||
|
|
||||||
|
```
|
||||||
|
adb push ./yolov5s_u8.rknn /userdata/rknn_yolo_demo/model/yolov5s_u8.rknn
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
## 单图测试执行
|
||||||
|
详情请了解官网：9.模型部署实例
|
||||||
|
|
||||||
|
进入EAI环境之后编译，将编译文件放置到可执行程序，假设是first_yolov5_detect_demo
|
||||||
|
用下列命令将可执行程序推送到开发板端
|
||||||
|
|
||||||
|
```
|
||||||
|
cp first_yolov5_detect_demo/ /mnt/userdata/ -rf
|
||||||
|
```
|
||||||
|
|
||||||
|
进入办卡运行环境
|
||||||
|
|
||||||
|
```
|
||||||
|
adb shell
|
||||||
|
```
|
||||||
|
|
||||||
|
定位到指定位置后，运行程序
|
||||||
|
|
||||||
|
```
|
||||||
|
./rknn_yolo_EAI
|
||||||
|
```
|
||||||
|
|
||||||
|
|
||||||
|
## 编译(build)
|
||||||
|
|
||||||
|
在项目文件中，根据设备，调整 GCC_COMPILER 参数，终端执行下面命令进行编译
|
||||||
|
|
||||||
|
```
|
||||||
|
./build.sh
|
||||||
|
```
|
||||||
47
build.sh
Normal file
47
build.sh
Normal file
@ -0,0 +1,47 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
rm build -rf
|
||||||
|
mkdir build
|
||||||
|
|
||||||
|
RV1109_TOOL_CHAIN="/opt/atk-dlrv1126-toolchain"
|
||||||
|
|
||||||
|
|
||||||
|
# for rv1109/rv1126 armhf
|
||||||
|
GCC_COMPILER=${RV1109_TOOL_CHAIN}/usr/bin/arm-linux-gnueabihf
|
||||||
|
|
||||||
|
ROOT_PWD=$( cd "$( dirname $0 )" && cd -P "$( dirname "$SOURCE" )" && pwd )
|
||||||
|
|
||||||
|
# model_zoo_path root path
|
||||||
|
MZ_ROOT=$(pwd | sed 's/\(rknn_model_zoo-main\).*/\1/g')
|
||||||
|
echo "MZ_ROOT:"
|
||||||
|
echo $MZ_ROOT
|
||||||
|
# build rockx
|
||||||
|
BUILD_DIR=${ROOT_PWD}/build
|
||||||
|
|
||||||
|
if [[ ! -d "${BUILD_DIR}" ]]; then
|
||||||
|
mkdir -p ${BUILD_DIR}
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd ${BUILD_DIR}
|
||||||
|
cmake .. \
|
||||||
|
-DCMAKE_C_COMPILER=${GCC_COMPILER}-gcc \
|
||||||
|
-DCMAKE_CXX_COMPILER=${GCC_COMPILER}-g++ \
|
||||||
|
-DMZ_ROOT=${MZ_ROOT} \
|
||||||
|
-DTARGET_SOC=RV1109_1126 \
|
||||||
|
-DCMAKE_SYSTEM_NAME=Linux
|
||||||
|
make -j4
|
||||||
|
make install
|
||||||
|
cd ..
|
||||||
|
|
||||||
|
cd ${BUILD_DIR}
|
||||||
|
cmake .. \
|
||||||
|
-DCMAKE_C_COMPILER=${GCC_COMPILER}-gcc \
|
||||||
|
-DCMAKE_CXX_COMPILER=${GCC_COMPILER}-g++ \
|
||||||
|
-DMZ_ROOT=${MZ_ROOT} \
|
||||||
|
-DTARGET_SOC=RV1109_1126 \
|
||||||
|
-DCMAKE_SYSTEM_NAME=Linux
|
||||||
|
make -j4
|
||||||
|
make install
|
||||||
|
|
||||||
Loading…
x
Reference in New Issue
Block a user