重新整理文件夹结构，并添加CNN训练代码。

2019-04-25 12:39:38 +08:00
parent 5c7317140a
commit ecbee8b24c
25 changed files with 325 additions and 19 deletions
--- a/others/src/options/options.cpp
+++ b/others/src/options/options.cpp
@@ -0,0 +1,68 @@
+//
+// Created by xinyang on 19-3-27.
+//
+
+#include <options/options.h>
+#include <log.h>
+#include <cstring>
+
+bool show_armor_box = false;
+bool show_armor_boxes = false;
+bool show_light_blobs = false;
+bool show_origin = false;
+bool save_labelled = false;
+bool run_with_camera = false;
+bool save_video = false;
+bool collect_data = false;
+
+void process_options(int argc, char *argv[]){
+    if(argc >= 2){
+        for(int i=1; i<argc; i++){
+            if(strcmp(argv[i], "--help") == 0){
+                LOGM("--show-armor-box: show the aim box.");
+                LOGM("--show-armor-boxes: show the candidate aim boxes.");
+                LOGM("--show-light-blobs: show the candidate light blobs.");
+                LOGM("--show-origin: show the origin image.");
+                LOGM("--save-label: save the image when box found.");
+                LOGM("--run-with-camera: start the program with camera directly without asking.");
+                LOGM("--save-video: save the video.");
+                LOGM("--collect-data: collect data sent from mcu.");
+            }else if(strcmp(argv[i], "--show-armor-box") == 0){
+                show_armor_box = true;
+                LOGM("Enable show armor box");
+            }else if(strcmp(argv[i], "--show-armor-boxes") == 0){
+                show_armor_boxes = true;
+                LOGM("Enable show armor boxes");
+            }else if(strcmp(argv[i], "--show-light-blobs") == 0) {
+                show_light_blobs = true;
+                LOGM("Enable show light blobs");
+            }else if(strcmp(argv[i], "--show-origin") == 0) {
+                show_origin = true;
+                LOGM("Enable show origin");
+            }else if(strcmp(argv[i], "--show-all") ==0 ) {
+                show_armor_box = true;
+                LOGM("Enable show armor box");
+                show_armor_boxes = true;
+                LOGM("Enable show armor boxes");
+                show_light_blobs = true;
+                LOGM("Enable show light blobs");
+                show_origin = true;
+                LOGM("Enable show origin");
+            }else if(strcmp(argv[i], "--save-labeled") == 0){
+                save_labelled = true;
+                LOGM("Enable save labeled");
+            }else if(strcmp(argv[i], "--run-with-camera") == 0){
+                run_with_camera = true;
+                LOGM("Run with camera!");
+            }else if(strcmp(argv[i], "--save-video") == 0){
+                save_video = true;
+                LOGM("Save video!");
+            }else if(strcmp(argv[i], "--collect-data") == 0){
+                collect_data = true;
+                LOGM("Enable data collection!");
+            }else{
+                LOGW("Unknown option: %s. Use --help to see options.", argv[i]);
+            }
+        }
+    }
+}