垃圾分类子项目-识别垃圾类型_垃圾识别代码-优快云博客

思路: 阿里云图像识别 + 语音模块实现

主程序:

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <unistd.h> // access()

#include <error.h> // remove()

#include "uartTool.h"

#include "garbage.h"

static int detect_process(const char *process_name) // 判断某个进程是否在运行

{

int n = -1;

FILE *strm;

char buf[128] = {0};

sprintf(buf, "ps -ax | grep %s|grep -v grep", process_name); // 组合进程名字，为完整命令

if ((strm = popen(buf, "r")) != NULL) // 通过 popen 的方式去执行

{

if (fgets(buf, sizeof(buf), strm) != NULL) // 执行完后判断是否能拿到正确的进程号，空格分开，第一个字符串就是进程号

{

n = atoi(buf); // 拿到就放回进程号，不然返回 -1

}

}

else

{

return -1; // 执行失败

}

pclose(strm);

return n;

}

int main(int argc, char **argv)

{

int serial_fd = -1;

int len = 0;

int ret = -1;

char *category = NULL;

unsigned char buffer[6] = {0xAA, 0x55, 0x00, 0x00, 0x55, 0xAA}; // 初始化 buffer[2] -- 关联垃圾类型

garbage_init(); // 先初始化阿里云接口

ret = detect_process("mjpg_streamer"); // 用于判断mjpg_streamer服务是否已经启动

if (-1 == ret)

{

goto END;

}

serial_fd = myserialOpen(SERIAL_DEV, BAUD); // 初始化串口,打开串口设备(语言模块)

if (serial_fd == -1)

{ // 初始化串口失败

goto END;

}

while (1) // 等待语言输入

{

len = serialGetstring(serial_fd, buffer); // 通过串口获得语言输入

printf("lend=%d, buf[2]=0x%x\n", len, buffer[2]);

if (len > 0 && buffer[2] == 0x46) // 判断是否需要启动识别

{

buffer[2] = 0x00;// 判断完后恢复，方便下次判断

system(WGET_CMD); // 拍照

//if (access(GARBAGE_FILE, F_OK) == 0) // 判断文件存在

// {

category = garbage_category(category); // 通过通过阿里云接口图像识别获取垃圾类型

if (strstr(category, "干垃圾"))

{

buffer[2] = 0x41;

}

else if (strstr(category, "湿垃圾"))

{

buffer[2] = 0x42;

}

else if (strstr(category, "可回收垃圾"))

{

buffer[2] = 0x43;

}

else if (strstr(category, "有害垃圾"))

{

buffer[2] = 0x44;

}

else

{

buffer[2] = 0x45;

}

//}

// else

//{ // 没有获取到图片

// buffer[2] = 0x45;

//}

serialSendstring(serial_fd, buffer, 6); // 将识别到的数据发送到串口

buffer[2] = 0x00; // 发送完后，一堆有效数据位清零，方便下一次调用

remove(GARBAGE_FILE); // 清理缓存删除刚刚拍摄的图片，避免对下一次拍摄造成干扰

}

}

END:

garbage_final();

return 0;

}

头文件

garbage.h

#ifndef __GARBAGE__H
#define __GARBAGE__H


void garbage_init(void);
void garbage_final(void);
char *garbage_category(char *category);

//为什么能用 127.0.0.1  我们是vscode 远程连接到香橙派 --  实际上是在orangepi上运行，用本地ip即可
#define WGET_CMD "wget http://127.0.0.1:8080/?action=snapshot -O/tmp/garbage.jpg"
#define GARBAGE_FILE "/tmp/garbage.jpg"

#endif

uartTool.h

#ifndef __UARTTOOL_H
#define _UARTTOOL_H

int myserialOpen (const char *device, const int baud);
void serialSendstring (const int fd, const unsigned char *s,int len);
int serialGetstring (const int fd,unsigned char *buffer);

//设置串口5  和 波特率 115200
#define SERIAL_DEV "/dev/ttyS5"
#define BAUD 115200

#endif

其他源文件:

garbage.c



#include<Python.h>
#include "garbage.h"

void garbage_init(void)
{
	Py_Initialize();
	PyObject *sys = PyImport_ImportModule("sys");

	PyObject *path = PyObject_GetAttrString(sys,"path");

	PyList_Append(path, PyUnicode_FromString("."));

}
void garbage_final(void)
{
         Py_Finalize();

}



char *garbage_category(char *category)
{

	PyObject *pModule= PyImport_ImportModule("garbage");
	if(!pModule)
	{
		PyErr_Print();
		printf("Error: failed to load garbage.py\n");
	        goto FAILED_MODULE;
	}

	PyObject *pfunc =PyObject_GetAttrString(pModule,"alibaba_garbage");


	if(!pfunc)
	{
		PyErr_Print();
		printf("Error: failed to load alibaba_garbage\n");
                goto FAILED_FUNC;
	}



	PyObject *pValue = PyObject_CallObject(pfunc, NULL);


	if(!pValue)
	{
		PyErr_Print();
		printf("Error:function call failed\n");
                goto FAILED_VALUE;

	}

	char * result = NULL;
	if(!PyArg_Parse(pValue,"s",&result))
	{

		PyErr_Print();
		printf("Error:garbage failed\n");
                goto FAILED_RESULT;

	}	
	printf("result=%s\n",result );
        // pValue 下面会被释放，他的result值就拿不到，得先存起来
        category = (char*)malloc(sizeof(char)*(strlen(result)+1));
        memset(category,0,(strlen(result)+1));
        strncpy(category,result,(strlen(result)+1));

	//注意释放顺序是反的，从新到旧
	// 添加跳转位置，当发生错误的时候，跳转过来将他释放
FAILED_RESULT:
	Py_DECREF(pValue);

FAILED_VALUE:
	Py_DECREF(pfunc);

FAILED_FUNC:
	Py_DECREF(pModule);

FAILED_MODULE:


	return  category;
}

uartTool.c

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <stdarg.h>
#include <string.h>
#include <termios.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "wiringSerial.h"
int myserialOpen (const char *device, const int baud)
{
struct termios options ;
speed_t myBaud ;
int status, fd ;
switch (baud){
case 9600: myBaud = B9600 ; break ;
case 115200: myBaud = B115200 ; break ;
}
if ((fd = open (device, O_RDWR | O_NOCTTY | O_NDELAY | O_NONBLOCK)) == -1)
return -1 ;
fcntl (fd, F_SETFL, O_RDWR) ;
// Get and modify current options:
tcgetattr (fd, &options) ;
cfmakeraw (&options) ;
cfsetispeed (&options, myBaud) ;
cfsetospeed (&options, myBaud) ;
options.c_cflag |= (CLOCAL | CREAD) ;
options.c_cflag &= ~PARENB ;
options.c_cflag &= ~CSTOPB ;
options.c_cflag &= ~CSIZE ;
options.c_cflag |= CS8 ;
options.c_lflag &= ~(ICANON | ECHO | ECHOE | ISIG) ;
options.c_oflag &= ~OPOST ;
options.c_cc [VMIN] = 0 ;
options.c_cc [VTIME] = 100 ; // Ten seconds (100 deciseconds)
tcsetattr (fd, TCSANOW, &options) ;
ioctl (fd, TIOCMGET, &status);
status |= TIOCM_DTR ;
status |= TIOCM_RTS ;
ioctl (fd, TIOCMSET, &status);
usleep (10000) ; // 10mS

return fd ;
}
//0xAA 0x55  0x64 00  0x55 0xAA  //0x AA > 157 unsigned
void serialSendstring (const int fd, const unsigned char *s,int len)
{
int ret;
ret = write (fd, s, len); //我们有结束位，而且不为'\0',所以不能用strlen
if (ret < 0)
printf("Serial Puts Error\n");
}
int serialGetstring (const int fd,unsigned char *buffer)
{
int n_read;
n_read = read(fd, buffer,32);
return n_read;
}