109 lines
2.7 KiB
C
109 lines
2.7 KiB
C
|
/*
|
||
|
|
Copyright (c), 2001-2022, Shenshu Tech. Co., Ltd.
|
||
|
|
*/
|
||
|
|
|
||
|
|
#include <stdio.h>
|
||
|
|
#include <stdlib.h>
|
||
|
|
#include <string.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include <pthread.h>
|
||
|
|
#include <signal.h>
|
||
|
|
#include <sys/types.h>
|
||
|
|
#include <sys/stat.h>
|
||
|
|
#include <fcntl.h>
|
||
|
|
#include <sys/ioctl.h>
|
||
|
|
|
||
|
|
#include "securec.h"
|
||
|
|
#include "sample_npu_process.h"
|
||
|
|
|
||
|
|
#define SAMPLE_SVP_NPU_ARG_MAX_NUM 3
|
||
|
|
#define SAMPLE_SVP_NPU_IDX_TWO 2
|
||
|
|
#define SAMPLE_SVP_NPU_CMP_STR_NUM 2
|
||
|
|
|
||
|
|
static char **g_npu_cmd_argv = TD_NULL;
|
||
|
|
|
||
|
|
/*
|
||
|
|
* function : to process abnormal case
|
||
|
|
*/
|
||
|
|
#ifndef __LITEOS__
|
||
|
|
static td_void sample_svp_npu_handle_sig(td_s32 signo)
|
||
|
|
{
|
||
|
|
if (signo == SIGINT || signo == SIGTERM) {
|
||
|
|
switch (*g_npu_cmd_argv[1]) {
|
||
|
|
case '0':
|
||
|
|
sample_svp_npu_acl_resnet50_handle_sig();
|
||
|
|
break;
|
||
|
|
case '1':
|
||
|
|
sample_svp_npu_acl_resnet50_handle_sig();
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
#endif
|
||
|
|
|
||
|
|
/*
|
||
|
|
* function : show usage
|
||
|
|
*/
|
||
|
|
static td_void sample_svp_npu_usage(const td_char *prg_name)
|
||
|
|
{
|
||
|
|
printf("Usage : %s <index>\n", prg_name);
|
||
|
|
printf("index:\n");
|
||
|
|
printf("\t 0) acl_resnet50.\n");
|
||
|
|
printf("\t 1) acl_resnet50_multithread.\n");
|
||
|
|
printf("\t 2) acl_mobilenet_v3_dynamicbatch_with_mmzcache.\n");
|
||
|
|
}
|
||
|
|
|
||
|
|
static td_s32 sample_svp_npu_case(int argc, char *argv[])
|
||
|
|
{
|
||
|
|
td_s32 ret = TD_SUCCESS;
|
||
|
|
|
||
|
|
switch (*argv[1]) {
|
||
|
|
case '0':
|
||
|
|
if (argc != SAMPLE_SVP_NPU_ARG_MAX_NUM - 1) {
|
||
|
|
return TD_FAILURE;
|
||
|
|
}
|
||
|
|
sample_svp_npu_acl_resnet50();
|
||
|
|
break;
|
||
|
|
case '1':
|
||
|
|
if (argc != SAMPLE_SVP_NPU_ARG_MAX_NUM - 1) {
|
||
|
|
return TD_FAILURE;
|
||
|
|
}
|
||
|
|
sample_svp_npu_acl_resnet50_multithread();
|
||
|
|
break;
|
||
|
|
case '2':
|
||
|
|
if (argc != SAMPLE_SVP_NPU_ARG_MAX_NUM - 1) {
|
||
|
|
return TD_FAILURE;
|
||
|
|
}
|
||
|
|
sample_svp_npu_acl_mobilenet_v3_dynamicbatch();
|
||
|
|
break;
|
||
|
|
default:
|
||
|
|
ret = TD_FAILURE;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
return ret;
|
||
|
|
}
|
||
|
|
|
||
|
|
#include "wrapperncnn.h"
|
||
|
|
|
||
|
|
// int main(int argc, char* argv[])
|
||
|
|
// {
|
||
|
|
// remove("image.png");
|
||
|
|
// struct sigaction sa;
|
||
|
|
// (td_void)memset_s(&sa, sizeof(struct sigaction), 0, sizeof(struct sigaction));
|
||
|
|
// sa.sa_handler = sample_svp_npu_handle_sig;
|
||
|
|
// sa.sa_flags = 0;
|
||
|
|
// sigaction(SIGTERM, &sa, NULL);
|
||
|
|
// if (argc != 2)
|
||
|
|
// {
|
||
|
|
// fprintf(stderr, "Usage: %s [imagepath]\n", argv[0]);
|
||
|
|
// return -1;
|
||
|
|
// }
|
||
|
|
// const char* imagepath = argv[1];
|
||
|
|
// ncnn_convertimg_yolov5s(imagepath, "testyuv.sp420");
|
||
|
|
// sample_svp_npu_acl_resnet50();
|
||
|
|
// system("chmod -R 777 image.png");
|
||
|
|
// return 0;
|
||
|
|
// }
|