差分

このページの2つのバージョン間の差分を表示します。

--- tutorial_3 [2022/01/04 14:20] – [第七項　スターターコードの完成] satoshi
+++ tutorial_3 [2022/01/05 10:49] (現在) – [第四項　Reactについて] satoshi
@@ 行 1: / 行 1: @@
-====== 第3章　スターターコード【中級編　ライン判定（通過判定）】 ======
+====== 第3章　ライン判定（WEBUIを使った通過判定）【中級編】 ======
 この章では、\\
   * エリア検知からラインをまたいだ検知への応用
@@ 行 265: / 行 265: @@
 等から学び始めることがよいでしょう。\\
-ただし今回はReact自体が分からなくても、どのプログラムファイルをどういじれば何が変わるかという点だけでピンポイントで解説していこうと思いますので、本チュートリアルを進めるうえではReactの学習は必須ではありません。\\
+ただし今回はReact自体が分からなくても、どのプログラムファイルをどういじれば何が変わるかという点だけでピンポイントで上級編の一環として解説していこうと思いますので、本チュートリアルを進めるうえではReactの学習は必須ではありません。\\
 ===== 第五項　WEBサーバーの解説=====
 WEBサーバーについては今回FlaskというPythonで簡単にWEBサーバーを構築できる仕組みを採用します。\\
@@ 行 565: / 行 566: @@
 ===== 第八項　エリア検知のUI連携 =====
-===== 第九項　カメラ座標から見取り図座標への切り替え =====
+ついでにエリア検知もUIで設定できるようにしましょう。変更箇所の解説はあまりいらないかと思いますので、結果プログラムを見てください。
+<sxh Python;title:03stopinfront_sketch.py>
+import sys
+import os
+import json
+import cv2
+import math
+from datetime import datetime, timedelta, timezone
+import csv
+import glob
+import numpy as np
+if __name__ == '__main__':
+    dirpath = 'data'
+    duration = 5
+    area_def_file = os.environ['HOME']+'/drawui/sketchareas.json'
+    JST = timezone(timedelta(hours=+9), 'JST')
+    print(sys.argv[1]+"を集計しています・・")
+    file_list = sorted(glob.glob('./'+dirpath+'/'+sys.argv[1]+'/raw_'+sys.argv[1]+'*.json'))
+    print("対象ファイル一覧")
+    for filename in file_list:
+        print(filename)
+    json_open = open(area_def_file, 'r')
+    result_json_file = open('result_sketchareas_'+sys.argv[1]+'.json', 'w')
+    areadef = json.load(json_open)
+    resultarr = {}
+    for areaidx, area in enumerate(areadef['areas']):
+        areaname='areaID'+str(areaidx)
+        print(areaname)
+        resultarr[areaname]={}
+    print("集計・・")
+    for areaidx, area in enumerate(areadef['areas']):
+        areaname='areaID'+str(areaidx)
+        print(areaname)
+        polygon = []
+        for point in area['points']:
+            polygon.append([point['x'],point['y']])
+        contour = np.array(polygon)
+        print(contour)
+        for filename in file_list:
+            print(areaname+"_"+filename)
+            if os.path.getsize(filename) == 0:
+                continue
+            raw_json_open = open(filename, 'r')
+            raw_json = json.load(raw_json_open)
+            for frame in raw_json:
+                jptime = datetime.fromtimestamp(frame['frame_time'],JST)
+                people = frame['objects']
+                for person in people:
+                    #print(areaname+'_'+person['tracking_id']+'_'+str(judgenum))
+                    personpos = [person["floor"]["x"],person["floor"]["y"]]
+                    test = cv2.pointPolygonTest(contour, personpos, False)
+                    if test>0:
+                        if person['tracking_id'] in resultarr[areaname]:
+                            resultarr[areaname][person['tracking_id']]['end_time'] = frame['frame_time']
+                            resultarr[areaname][person['tracking_id']]['duration'] = frame['frame_time']-resultarr[areaname][person['tracking_id']]['start_time']
+                            resultarr[areaname][person['tracking_id']]['end_time_jp'] = jptime.isoformat()
+                            if resultarr[areaname][person['tracking_id']]['duration'] > duration:
+                                resultarr[areaname][person['tracking_id']]['stop_judge'] = True
+                        else:
+                            resultarr[areaname][person['tracking_id']]={'area':areaname,'start_time_jp':jptime.isoformat(),'end_time_jp':jptime.isoformat(),'start_time':frame['frame_time'],'end_time':frame['frame_time'],'stream_id':frame['stream_id'],'duration':0,'stop_judge':False}
+    json.dump(resultarr,result_json_file)
+    for areaname, records in resultarr.items():
+        print(areaname)
+        f = open("sketchareas_"+sys.argv[1]+"_"+areaname+".csv", 'w', encoding='utf-8', newline='')
+        write = csv.writer(f)
+        write.writerow(["stream_id","areaname","tracking_id","start_time_jp","end_time_jp","total_duration","stop_judge"])
+        for tracking_id, record in records.items():
+            write.writerow([record['stream_id'],areaname,tracking_id,record['start_time_jp'],record['end_time_jp'],record['duration'],record['stop_judge']])
+        f.close()
+</sxh>