dockerapi

2021-06-09 08:27:32 +00:00
parent bd0c692628
commit 097c38a109
15 changed files with 12408 additions and 27 deletions
@@ -0,0 +1,69 @@
+cmake_minimum_required(VERSION 3.5)
+set(PROJ_NAME tkDNN)
+project (tkDNN)
+set(CMAKE_MODULE_PATH ${CMAKE_MODULE_PATH} ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
+if(UNIX)
+
+####
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11 -fPIC -Wno-deprecated-declarations -Wno-unused-variable ")
+endif()
+if(WIN32)
+set(CMAKE_CXX_STANDARD 11)
+set(CMAKE_CXX_FLAGS "/O2 /FS /EHsc")
+set(CMAKE_WINDOWS_EXPORT_ALL_SYMBOLS ON)
+#add extras for baggage
+endif(WIN32)
+include_directories(${CMAKE_CURRENT_SOURCE_DIR}/include/tkDNN)
+
+# project specific flags
+if(DEBUG)
+    add_definitions(-DDEBUG)
+endif()
+
+add_definitions(-DTKDNN_PATH="${CMAKE_CURRENT_SOURCE_DIR}")
+
+#-------------------------------------------------------------------------------
+# CUDA
+#-------------------------------------------------------------------
+#-------------------------------------------------------------------------------
+# External Libraries
+#------------------------------------------------------------------------------
+find_package(OpenCV REQUIRED)
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -DOPENCV")
+
+# gives pr
+#-------------------------------------------------------------------------------
+# Build Libraries
+#----------------------------------------------------------------------------
+file(GLOB_RECURSE SOURCE_FILES "client_img.cpp")
+
+add_executable(client ${SOURCE_FILES})
+#set(Casablanca_LIBRARIES "-lboost_log -lboost_log_setup -lboost_thread -lboost_system -lcrypto -lssl -lcpprest -lpthread" )
+set(tkdnn_LIBS ${OpenCV_LIBS})
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
+# Link BaggageAI library' include folder.
+include_directories(${OPENCV_INCLUDE_DIRS} ${CMAKE_CXX_FLAGS})
+# Define BaggageAI library' shared library.
+#add_library(${BAGGAGEAI_LIB_NAME} SHARED IMPORTED)
+# Set a path to BaggageAI library' shared library
+#set_property(TARGET ${BAGGAGEAI_LIB_NAME} PROPERTY IMPORTED_LOCATION "${BAGGAGEAI_PATH}/libBaggageAI.so")
+
+# Link all libraries together.
+target_link_libraries(client ${tkdnn_LIBS})
+
+#static 
+#add_executable(demo demo/inf.cpp)
+#target_link_libraries(demo tkDNN)
+
+#-------------------------------------------------------------------------------
+# Install
+#-------------------------------------------------------------------------------
+#if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
+#    set (CMAKE_INSTALL_PREFIX "${CMAKE_BINARY_DIR}/install" 
+#         CACHE PATH "default install path" FORCE)
+#endif()
+message("install dir:" ${CMAKE_INSTALL_PREFIX})
+install(DIRECTORY "${CMAKE_CURRENT_SOURCE_DIR}/cmake/" # source directory
+        DESTINATION "share/tkDNN/cmake/" # target directory
+)
+
@@ -0,0 +1,13 @@
+DEBUG= -g
+OPENCV= -lopencv_core -lopencv_highgui -lopencv_imgproc -lopencv_video
+
+PKG_CONF_CFLAG= 'pkg-config --cflags opencv --libs opencv'
+PKGCONF_LIBS= 'pkg-config --libs opencv'
+
+client:
+	clear
+	g++ -std=c++11 $(PKG_CONF_CFLAG) -o client client.cpp $(PKGCONF_LIBS)
+
+server:
+	clear
+	g++ -std=c++11 $(PKG_CONF_CFLAG) -o server server.cpp $(PKGCONF_LIBS)
@@ -0,0 +1,89 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <netdb.h>
+#include <iostream>
+#include "opencv2/core/core.hpp"
+#include <opencv2/highgui/highgui.hpp>
+#include <opencv2/imgproc/imgproc.hpp>
+
+#define LOCALHOST "127.0.0.1"
+#define PORT 8080
+#define FRAME_WIDTH         608
+#define FRAME_HEIGHT        608
+
+using namespace std;
+using namespace cv;
+
+void error(const char *msg)
+{
+    perror(msg);
+    exit(0);
+}
+
+int main()
+{
+  int sockfd, portno, n, imgSize, IM_HEIGHT, IM_WIDTH;
+  struct sockaddr_in serv_addr;
+  struct hostent *server;
+  char buffer[256];
+  Mat cameraFeed;
+
+  portno = PORT;
+  sockfd = socket(AF_INET, SOCK_STREAM, 0);
+
+  if (sockfd < 0) error("ERROR opening socket");
+
+  server = gethostbyname(LOCALHOST);
+
+  if (server == NULL) {
+      fprintf(stderr,"ERROR, no such host\n");
+      exit(0);
+  }
+
+  bzero((char *) &serv_addr, sizeof(serv_addr));
+  serv_addr.sin_family = AF_INET;
+  bcopy((char *)server->h_addr,
+       (char *)&serv_addr.sin_addr.s_addr,
+       server->h_length);
+  serv_addr.sin_port = htons(portno);
+
+  if (connect(sockfd,(struct sockaddr *) &serv_addr,sizeof(serv_addr)) < 0)
+      error("ERROR connecting");
+
+//  VideoCapture capture;
+
+//	capture.open(0);
+
+ // while(true)
+ // {
+    /* store image to matrix && test frame */
+//    cv::Mat frame=cv::imread(demo.jpg);
+    cameraFeed = cv::imread("demo.jpg", cv::IMREAD_COLOR);
+    int height = cameraFeed.rows;
+    int width = cameraFeed.cols;
+   std::cout<<width;
+   // Mat cropped = Mat(cameraFeed, Rect(width/2 - width/7,
+//                                       height/2 - height/9,
+  //                                     2*width/7, 2*height/7));
+//    cameraFeed = frame;
+
+    IM_HEIGHT = FRAME_HEIGHT;
+    IM_WIDTH = FRAME_WIDTH;
+
+    resize(cameraFeed, cameraFeed, Size( IM_WIDTH , IM_HEIGHT ));
+
+    imgSize=cameraFeed.total()*cameraFeed.elemSize();
+    n = send(sockfd, cameraFeed.data, imgSize, 0);
+    std::cout<<n;
+    if (n < 0) error("ERROR writing to socket");
+  //}
+
+  close(sockfd);
+
+  return 0;
+}
@@ -0,0 +1,102 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <iostream>
+#include "opencv2/core/core.hpp"
+#include <opencv2/highgui/highgui.hpp>
+
+#include<sys/socket.h>	//socket
+#include<sys/types.h>
+#include<netinet/in.h>
+
+using namespace std;
+using namespace cv;
+
+#define PORT 7200
+
+#define FRAME_WIDTH         640
+#define FRAME_HEIGHT        480
+
+void error(const char *msg)
+{
+  perror(msg);
+  exit(1);
+}
+
+int main()
+{
+  int sockfd, newsockfd, portno, n, imgSize, bytes=0, IM_HEIGHT, IM_WIDTH;;
+  socklen_t clilen;
+  char buffer[256];
+  struct sockaddr_in serv_addr, cli_addr;
+  Mat img;
+
+  sockfd=socket(AF_INET, SOCK_STREAM, 0);
+  if(sockfd<0) error("ERROR opening socket");
+
+  bzero((char*)&serv_addr, sizeof(serv_addr));
+  portno = PORT;
+
+  serv_addr.sin_family=AF_INET;
+  serv_addr.sin_addr.s_addr=INADDR_ANY;
+  serv_addr.sin_port=htons(portno);
+
+  if(bind(sockfd, (struct sockaddr *) &serv_addr,
+          sizeof(serv_addr))<0) error("ERROR on binding");
+
+  listen(sockfd,5);
+  clilen=sizeof(cli_addr);
+
+  newsockfd=accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
+  if(newsockfd<0) error("ERROR on accept");
+
+  uchar sock[3];
+  cout << sock <<endl;
+  cout << sock+3 <<endl;
+
+  // bzero(buffer,1024);
+  // n = read(newsockfd, buffer, 1023);
+  // if(n<0) error("ERROR reading from socket");
+  //printf("Here is the message: %s\n", buffer);
+
+  // n=write(newsockfd, "I got your message", 18);
+  // if(n<0) error("ERROR writing to socket");
+  bool running = true;
+
+  while(running)
+  { std::cout<<"t"<<"\n";
+    IM_HEIGHT = FRAME_HEIGHT;
+    IM_WIDTH = FRAME_WIDTH;
+    img = Mat::zeros(FRAME_HEIGHT, FRAME_WIDTH, CV_8UC3);
+
+    imgSize = img.total()*img.elemSize();
+    uchar sockData[imgSize];
+    std::cout<<"t2"<<"\n";
+    for(int i=0;i<imgSize;i+=bytes)
+      if ((bytes=recv(newsockfd, sockData+i, imgSize-i,0))==-1) error("recv failed");
+
+    int ptr=0;
+
+    for(int i=0;i<img.rows;++i)
+      for(int j=0;j<img.cols;++j)
+      {
+        img.at<Vec3b>(i,j) = Vec3b(sockData[ptr+0],sockData[ptr+1],sockData[ptr+2]);
+        ptr=ptr+3;
+      }
+    std::cout<<"t3"<<"\n";
+    int height = img.cols;
+    std::cout<<height;
+//    namedWindow( "Server", CV_WINDOW_AUTOSIZE );// Create a window for display.
+  //  imshow( "Server", img );
+    char key = waitKey(30);
+    running = key;
+    //esc
+    if(key==27) running =false;
+  }
+
+  close(newsockfd);
+  close(sockfd);
+
+  return 0;
+}