mirror of https://github.com/AlexeyAB/darknet.git
AlexeyAB
7 years ago
parent
baf3fcb491
commit
82f630cac7
13 changed files with 265 additions and 11 deletions
@ -0,0 +1,7 @@ |
||||
rem Run this file and then open URL in Chrome/Firefox: rem http://localhost:8090 |
||||
rem Or open: http://ip-address:8090 |
||||
|
||||
darknet.exe detector demo data/voc.data yolo-voc.cfg yolo-voc.weights test.mp4 -i 0 -http_port 8090 |
||||
|
||||
|
||||
pause |
||||
@ -0,0 +1,8 @@ |
||||
rem Run this file and then open URL in Chrome/Firefox: rem http://localhost:8090 |
||||
rem Or open: http://ip-address:8090 |
||||
|
||||
./darknet detector demo ./cfg/voc.data ./cfg/yolo-voc.cfg ./yolo-voc.weights test50.mp4 -i 0 -thresh 0.2 -http_port 8090 |
||||
|
||||
|
||||
|
||||
pause |
||||
@ -0,0 +1,197 @@ |
||||
#ifdef OPENCV |
||||
//
|
||||
// a single-threaded, multi client(using select), debug webserver - streaming out mjpg.
|
||||
// on win, _WIN32 has to be defined, must link against ws2_32.lib (socks on linux are for free)
|
||||
//
|
||||
|
||||
//
|
||||
// socket related abstractions:
|
||||
//
|
||||
#ifdef _WIN32 |
||||
#pragma comment(lib, "ws2_32.lib") |
||||
#include <winsock.h> |
||||
#include <windows.h> |
||||
#include <time.h> |
||||
#define PORT unsigned long |
||||
#define ADDRPOINTER int* |
||||
struct _INIT_W32DATA |
||||
{ |
||||
WSADATA w; |
||||
_INIT_W32DATA() { WSAStartup(MAKEWORD(2, 1), &w); } |
||||
} _init_once; |
||||
#else /* ! win32 */ |
||||
#include <unistd.h> |
||||
#include <sys/time.h> |
||||
#include <sys/types.h> |
||||
#include <sys/socket.h> |
||||
#include <netdb.h> |
||||
#include <netinet/in.h> |
||||
#include <arpa/inet.h> |
||||
#define PORT unsigned short |
||||
#define SOCKET int |
||||
#define HOSTENT struct hostent |
||||
#define SOCKADDR struct sockaddr |
||||
#define SOCKADDR_IN struct sockaddr_in |
||||
#define ADDRPOINTER unsigned int* |
||||
#define INVALID_SOCKET -1 |
||||
#define SOCKET_ERROR -1 |
||||
#endif /* _WIN32 */ |
||||
|
||||
#include <vector> |
||||
#include <iostream> |
||||
using std::cerr; |
||||
using std::endl; |
||||
|
||||
#include "opencv2/opencv.hpp" |
||||
using namespace cv; |
||||
|
||||
#include "http_stream.h" |
||||
|
||||
|
||||
class MJPGWriter |
||||
{ |
||||
SOCKET sock; |
||||
SOCKET maxfd; |
||||
fd_set master; |
||||
int timeout; // master sock timeout, shutdown after timeout millis.
|
||||
int quality; // jpeg compression [1..100]
|
||||
|
||||
int _write(int sock, char *s, int len) |
||||
{ |
||||
if (len < 1) { len = strlen(s); } |
||||
return ::send(sock, s, len, 0); |
||||
} |
||||
|
||||
public: |
||||
|
||||
MJPGWriter(int port = 0, int _timeout = 200000, int _quality = 30) |
||||
: sock(INVALID_SOCKET) |
||||
, timeout(_timeout) |
||||
, quality(_quality) |
||||
{ |
||||
FD_ZERO(&master); |
||||
if (port) |
||||
open(port); |
||||
} |
||||
|
||||
~MJPGWriter() |
||||
{ |
||||
release(); |
||||
} |
||||
|
||||
bool release() |
||||
{ |
||||
if (sock != INVALID_SOCKET) |
||||
::shutdown(sock, 2); |
||||
sock = (INVALID_SOCKET); |
||||
return false; |
||||
} |
||||
|
||||
bool open(int port) |
||||
{ |
||||
sock = ::socket(AF_INET, SOCK_STREAM, IPPROTO_TCP); |
||||
|
||||
SOCKADDR_IN address; |
||||
address.sin_addr.s_addr = INADDR_ANY; |
||||
address.sin_family = AF_INET; |
||||
address.sin_port = htons(port); // ::htons(port);
|
||||
if (::bind(sock, (SOCKADDR*)&address, sizeof(SOCKADDR_IN)) == SOCKET_ERROR) |
||||
{ |
||||
cerr << "error : couldn't bind sock " << sock << " to port " << port << "!" << endl; |
||||
return release(); |
||||
} |
||||
if (::listen(sock, 10) == SOCKET_ERROR) |
||||
{ |
||||
cerr << "error : couldn't listen on sock " << sock << " on port " << port << " !" << endl; |
||||
return release(); |
||||
} |
||||
FD_ZERO(&master); |
||||
FD_SET(sock, &master); |
||||
maxfd = sock; |
||||
return true; |
||||
} |
||||
|
||||
bool isOpened() |
||||
{ |
||||
return sock != INVALID_SOCKET; |
||||
} |
||||
|
||||
bool write(const Mat & frame) |
||||
{ |
||||
fd_set rread = master; |
||||
struct timeval to = { 0,timeout }; |
||||
if (::select(maxfd+1, &rread, NULL, NULL, &to) <= 0) |
||||
return true; // nothing broken, there's just noone listening
|
||||
|
||||
std::vector<uchar> outbuf; |
||||
std::vector<int> params; |
||||
params.push_back(IMWRITE_JPEG_QUALITY); |
||||
params.push_back(quality); |
||||
cv::imencode(".jpg", frame, outbuf, params); |
||||
unsigned int outlen = outbuf.size(); |
||||
|
||||
#ifdef _WIN32 |
||||
for (unsigned i = 0; i<rread.fd_count; i++) |
||||
{ |
||||
int addrlen = sizeof(SOCKADDR); |
||||
SOCKET s = rread.fd_array[i]; // fd_set on win is an array, while ...
|
||||
#else |
||||
for (int s = 0; s<=maxfd; s++) |
||||
{ |
||||
socklen_t addrlen = sizeof(SOCKADDR); |
||||
if (!FD_ISSET(s, &rread)) // ... on linux it's a bitmask ;)
|
||||
continue; |
||||
#endif |
||||
if (s == sock) // request on master socket, accept and send main header.
|
||||
{ |
||||
SOCKADDR_IN address = { 0 }; |
||||
SOCKET client = ::accept(sock, (SOCKADDR*)&address, &addrlen); |
||||
if (client == SOCKET_ERROR) |
||||
{ |
||||
cerr << "error : couldn't accept connection on sock " << sock << " !" << endl; |
||||
return false; |
||||
} |
||||
maxfd = (maxfd>client ? maxfd : client); |
||||
FD_SET(client, &master); |
||||
_write(client, "HTTP/1.0 200 OK\r\n", 0); |
||||
_write(client, |
||||
"Server: Mozarella/2.2\r\n" |
||||
"Accept-Range: bytes\r\n" |
||||
"Connection: close\r\n" |
||||
"Max-Age: 0\r\n" |
||||
"Expires: 0\r\n" |
||||
"Cache-Control: no-cache, private\r\n" |
||||
"Pragma: no-cache\r\n" |
||||
"Content-Type: multipart/x-mixed-replace; boundary=mjpegstream\r\n" |
||||
"\r\n", 0); |
||||
cerr << "new client " << client << endl; |
||||
} |
||||
else // existing client, just stream pix
|
||||
{ |
||||
char head[400]; |
||||
sprintf(head, "--mjpegstream\r\nContent-Type: image/jpeg\r\nContent-Length: %lu\r\n\r\n", outlen); |
||||
_write(s, head, 0); |
||||
int n = _write(s, (char*)(&outbuf[0]), outlen); |
||||
//cerr << "known client " << s << " " << n << endl;
|
||||
if (n < outlen) |
||||
{ |
||||
cerr << "kill client " << s << endl; |
||||
::shutdown(s, 2); |
||||
FD_CLR(s, &master); |
||||
} |
||||
} |
||||
} |
||||
return true; |
||||
} |
||||
}; |
||||
|
||||
|
||||
|
||||
void send_mjpeg(IplImage* ipl, int port, int timeout, int quality) { |
||||
static MJPGWriter wri(port, timeout, quality); |
||||
cv::Mat mat = cv::cvarrToMat(ipl); |
||||
wri.write(mat); |
||||
std::cout << " MJPEG-stream sent. \n"; |
||||
} |
||||
|
||||
#endif // OPENCV
|
||||
@ -0,0 +1,15 @@ |
||||
#pragma once |
||||
#ifndef HTTP_STREAM_H |
||||
#define HTTP_STREAM_H |
||||
|
||||
#ifdef __cplusplus |
||||
extern "C" { |
||||
#endif |
||||
|
||||
void send_mjpeg(IplImage* ipl, int port, int timeout, int quality); |
||||
|
||||
#ifdef __cplusplus |
||||
} |
||||
#endif |
||||
|
||||
#endif // HTTP_STREAM_H
|
||||
Loading…
Reference in new issue