Skip to content

J
Jetson-Triton-LPR
Commit bd42474b authored by YONG-LIN SU's avatar YONG-LIN SU
Browse files
Options

20220621 modify camera queue

parents d03d2cf2 d057d3bc
Hide whitespace changes
Inline Side-by-side
Showing with 286 additions and 10 deletions
.ipynb_checkpoints/test_camera-checkpoint.ipynb
View file @ bd42474b
......@@ -178,7 +178,18 @@
"name": "stdout",
"output_type": "stream",
"text": [
"30.0\n"
"60.0\n"
]
},
{
"ename": "KeyboardInterrupt",
"evalue": "",
"output_type": "error",
"traceback": [
"\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
"\u001b[0;31mKeyboardInterrupt\u001b[0m Traceback (most recent call last)",
"\u001b[0;32m<ipython-input-2-4323a410c5ab>\u001b[0m in \u001b[0;36m<module>\u001b[0;34m\u001b[0m\n\u001b[1;32m 32\u001b[0m \u001b[0;32mif\u001b[0m \u001b[0mret\u001b[0m\u001b[0;34m:\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 33\u001b[0m \u001b[0;31m#VioWri.Frame = frame\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0;32m---> 34\u001b[0;31m \u001b[0mwriter\u001b[0m\u001b[0;34m.\u001b[0m\u001b[0mwrite\u001b[0m\u001b[0;34m(\u001b[0m\u001b[0mframe\u001b[0m\u001b[0;34m)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[0m\u001b[1;32m 35\u001b[0m \u001b[0;31m# cv2.imshow(\"1\",frame)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n\u001b[1;32m 36\u001b[0m \u001b[0;31m# key=cv2.waitKey(1)\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0;34m\u001b[0m\u001b[0m\n",
"\u001b[0;31mKeyboardInterrupt\u001b[0m: "
]
}
],
......@@ -511,9 +522,131 @@
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 2,
"id": "78266468",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Saved: Kevin_Car_i_frame_0.jpg\n",
"Saved: Kevin_Car_i_frame_30.jpg\n",
"Saved: Kevin_Car_i_frame_60.jpg\n",
"Saved: Kevin_Car_i_frame_90.jpg\n",
"Saved: Kevin_Car_i_frame_120.jpg\n",
"Saved: Kevin_Car_i_frame_150.jpg\n",
"Saved: Kevin_Car_i_frame_180.jpg\n",
"Saved: Kevin_Car_i_frame_210.jpg\n",
"Saved: Kevin_Car_i_frame_240.jpg\n",
"Saved: Kevin_Car_i_frame_270.jpg\n",
"Saved: Kevin_Car_i_frame_300.jpg\n",
"Saved: Kevin_Car_i_frame_330.jpg\n",
"Saved: Kevin_Car_i_frame_360.jpg\n",
"Saved: Kevin_Car_i_frame_390.jpg\n",
"Saved: Kevin_Car_i_frame_420.jpg\n",
"Saved: Kevin_Car_i_frame_450.jpg\n"
]
}
],
"source": [
"# FFmpeg 取關鍵禎 I-Frame\n",
"\n",
"import os\n",
"import cv2\n",
"import subprocess\n",
"\n",
"filename = '/home/tx2/Videos/Kevin_Car.MOV'\n",
"\n",
"def get_frame_types(video_fn):\n",
" command = 'ffprobe -v error -show_entries frame=pict_type -of default=noprint_wrappers=1'.split()\n",
" out = subprocess.check_output(command + [video_fn]).decode()\n",
" frame_types = out.replace('pict_type=','').split()\n",
" return zip(range(len(frame_types)), frame_types)\n",
"\n",
"def save_i_keyframes(video_fn):\n",
" frame_types = get_frame_types(video_fn)\n",
" i_frames = [x[0] for x in frame_types if x[1]=='I']\n",
" if i_frames:\n",
" basename = os.path.splitext(os.path.basename(video_fn))[0]\n",
" cap = cv2.VideoCapture(video_fn)\n",
" for frame_no in i_frames:\n",
" cap.set(cv2.CAP_PROP_POS_FRAMES, frame_no)\n",
" ret, frame = cap.read()\n",
" outname = basename+'_i_frame_'+str(frame_no)+'.jpg'\n",
" cv2.imwrite(outname, frame)\n",
" print ('Saved: '+outname)\n",
" cap.release()\n",
" else:\n",
" print ('No I-frames in '+video_fn)\n",
"\n",
"if __name__ == '__main__':\n",
" save_i_keyframes(filename)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "e5b68fa5",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"29.97002997002997\n"
]
}
],
"source": [
"# video writer per hour\n",
"import cv2\n",
"\n",
"source=\"rtsp://192.168.5.218/txg/01\"\n",
"\n",
"gst = \"filesrc location=/home/tx2/Videos/Kevin_Car.MOV ! qtdemux ! queue ! h264parse ! nvv4l2decoder ! nvvidconv ! video/x-raw,format=BGRx ! queue ! videoconvert ! queue ! video/x-raw, format=BGR ! appsink\"\n",
"\n",
"cap=cv2.VideoCapture(gst,cv2.CAP_GSTREAMER)\n",
"\n",
"\n",
"fps=cap.get(cv2.CAP_PROP_FPS)\n",
"print(fps)\n",
"\n",
"break_flag=False\n",
"try:\n",
" cv2.namedWindow(\"1\",cv2.WINDOW_NORMAL)\n",
" cv2.resizeWindow(\"1\",1280,720)\n",
" while True:\n",
" if break_flag:\n",
" break\n",
" ret,frame=cap.read()\n",
" if ret:\n",
" cv2.imshow(\"1\",frame)\n",
" key=cv2.waitKey(33)\n",
" if key==ord('q'):\n",
" break_flag=True\n",
" break\n",
"except Exception as e:\n",
" raise(e)\n",
" cap.release()\n",
" cv2.destroyAllWindows()\n",
" \n",
"finally:\n",
" cap.release()\n",
" cv2.destroyAllWindows()\n",
" \n",
" \n",
"\n",
" \n",
" \n",
" "
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "8befd1ac",
"metadata": {},
"outputs": [],
"source": []
}
......
PlanB_test0609.py
View file @ bd42474b
......@@ -30,12 +30,12 @@ yolo = TrtYOLOSHM()
easyLPR=EZLPR()
#source="/home/tx2/Videos/20220504173000.mp4"
source="/home/tx2/Videos/Kevin_Car.MOV"
#source="/home/tx2/Videos/Kevin_Car.MOV"
#source = "../Ch17-20220314-190000.mp4"
#source="rtsp://192.168.5.218/txg/01"
source="rtsp://192.168.5.218/txg/01"
#source="rtsp://140.120.65.31/live.sdp"
cam=Camera(1,source,encoder = "h265",width=1920,height=1080)
cam=Camera(1,source,encoder = "h265",width=1920,height=1080,skip_frames=0)
camera_id=1
break_flag=False
......@@ -104,7 +104,7 @@ try:
TFMeta.UpdateTrafficFlow(objects)
TF_span,stime=time_span(stime) # !
if is_display:
if is_display and False:
displayMeta.frame = TFMeta.osd(frame) # 車流判斷的疊圖很耗時可考慮註解
for index,obj in objects.items():
# draw trajectory
......@@ -128,7 +128,6 @@ try:
CT.objects[index].update_lpr_candis(platNum)
OutputLPR_span,stime=time_span(stime) # !
CT.GetDisposeBuffer()
for obj in CT.GetDisposeBuffer():
pass
# InfluxWriter.put(obj)
......@@ -155,7 +154,8 @@ try:
# f"Influx_Buffer:{InfluxWriter.ObjectMetaUploadQueue.qsize()}")
if is_display:
cv2.imshow("1",displayMeta.frame)
key=cv2.waitKey(1)
waittime= 33 - int(total_span*1000) if 33 - int(total_span*1000) > 0 else 1
key=cv2.waitKey(waittime)
if key==ord('q'):
break_flag=True
break
......
VideoAnalysis/Camera_Jetson.py
View file @ bd42474b
......@@ -25,7 +25,7 @@ class Camera(object):
use_gstr: 是否使用gstreamer後台
skip_frames: 是否跳過部分幀，1 => 全部解碼， 2 => 只解碼 non-ref frames，3 => 只解碼 key-frames
skip_frames: 是否跳過部分幀，0 => 全部解碼， 1 => 只解碼 non-ref frames，2 => 只解碼 key-frames
"""
......
test_camera.ipynb
View file @ bd42474b
......@@ -522,9 +522,152 @@
},
{
"cell_type": "code",
"execution_count": null,
"execution_count": 2,
"id": "78266468",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"Saved: Kevin_Car_i_frame_0.jpg\n",
"Saved: Kevin_Car_i_frame_30.jpg\n",
"Saved: Kevin_Car_i_frame_60.jpg\n",
"Saved: Kevin_Car_i_frame_90.jpg\n",
"Saved: Kevin_Car_i_frame_120.jpg\n",
"Saved: Kevin_Car_i_frame_150.jpg\n",
"Saved: Kevin_Car_i_frame_180.jpg\n",
"Saved: Kevin_Car_i_frame_210.jpg\n",
"Saved: Kevin_Car_i_frame_240.jpg\n",
"Saved: Kevin_Car_i_frame_270.jpg\n",
"Saved: Kevin_Car_i_frame_300.jpg\n",
"Saved: Kevin_Car_i_frame_330.jpg\n",
"Saved: Kevin_Car_i_frame_360.jpg\n",
"Saved: Kevin_Car_i_frame_390.jpg\n",
"Saved: Kevin_Car_i_frame_420.jpg\n",
"Saved: Kevin_Car_i_frame_450.jpg\n"
]
}
],
"source": [
"# FFmpeg 取關鍵禎 I-Frame\n",
"\n",
"import os\n",
"import cv2\n",
"import subprocess\n",
"\n",
"filename = '/home/tx2/Videos/Kevin_Car.MOV'\n",
"\n",
"def get_frame_types(video_fn):\n",
" command = 'ffprobe -v error -show_entries frame=pict_type -of default=noprint_wrappers=1'.split()\n",
" out = subprocess.check_output(command + [video_fn]).decode()\n",
" frame_types = out.replace('pict_type=','').split()\n",
" return zip(range(len(frame_types)), frame_types)\n",
"\n",
"def save_i_keyframes(video_fn):\n",
" frame_types = get_frame_types(video_fn)\n",
" i_frames = [x[0] for x in frame_types if x[1]=='I']\n",
" if i_frames:\n",
" basename = os.path.splitext(os.path.basename(video_fn))[0]\n",
" cap = cv2.VideoCapture(video_fn)\n",
" for frame_no in i_frames:\n",
" cap.set(cv2.CAP_PROP_POS_FRAMES, frame_no)\n",
" ret, frame = cap.read()\n",
" outname = basename+'_i_frame_'+str(frame_no)+'.jpg'\n",
" cv2.imwrite(outname, frame)\n",
" print ('Saved: '+outname)\n",
" cap.release()\n",
" else:\n",
" print ('No I-frames in '+video_fn)\n",
"\n",
"if __name__ == '__main__':\n",
" save_i_keyframes(filename)"
]
},
{
"cell_type": "code",
"execution_count": 6,
"id": "e5b68fa5",
"metadata": {},
"outputs": [
{
"name": "stdout",
"output_type": "stream",
"text": [
"29.97002997002997\n"
]
}
],
"source": [
"# video writer per hour\n",
"import cv2\n",
"\n",
"source=\"rtsp://192.168.5.218/txg/01\"\n",
"\n",
"gst = \"filesrc location=/home/tx2/Videos/Kevin_Car.MOV ! qtdemux ! queue ! h264parse ! nvv4l2decoder ! nvvidconv ! video/x-raw,format=BGRx ! queue ! videoconvert ! queue ! video/x-raw, format=BGR ! appsink\"\n",
"\n",
"cap=cv2.VideoCapture(gst,cv2.CAP_GSTREAMER)\n",
"\n",
"\n",
"fps=cap.get(cv2.CAP_PROP_FPS)\n",
"print(fps)\n",
"\n",
"break_flag=False\n",
"try:\n",
" cv2.namedWindow(\"1\",cv2.WINDOW_NORMAL)\n",
" cv2.resizeWindow(\"1\",1280,720)\n",
" while True:\n",
" if break_flag:\n",
" break\n",
" ret,frame=cap.read()\n",
" if ret:\n",
" cv2.imshow(\"1\",frame)\n",
" key=cv2.waitKey(33)\n",
" if key==ord('q'):\n",
" break_flag=True\n",
" break\n",
"except Exception as e:\n",
" raise(e)\n",
" cap.release()\n",
" cv2.destroyAllWindows()\n",
" \n",
"finally:\n",
" cap.release()\n",
" cv2.destroyAllWindows()\n",
" \n",
" \n",
"\n",
" \n",
" \n",
" "
]
},
{
"cell_type": "code",
"execution_count": 9,
"id": "3f7ad718",
"metadata": {},
"outputs": [
{
"data": {
"text/plain": [
"35"
]
},
"execution_count": 9,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"int(0.035 *1000)"
]
},
{
"cell_type": "code",
"execution_count": null,
"id": "b964b08c",
"metadata": {},
"outputs": [],
"source": []
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment