Added source option

Added another source option for custom links to H265 RTSP sources

Author: leo-jolibois

App/Demo_squelet/ia_squelet.py

@@ -54,9 +54,12 @@ def switch_case(argument):
 		case 'prompt_ip_webcam':
 			print("Please enter the device's IP address")
 			return 'rtspsrc location=rtsp://'+input()+':8080/h264_ulaw.sdp latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! videoscale ! video/x-raw,width=1280,height=720,format=BGR ! appsink drop=1'
-		case 'prompt_rtsp':
+		case 'prompt_rtsp_h264':
 			print("Please enter the full link")
-			return  'rtspsrc location='+input()+' latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! videoscale ! video/x-raw,width=1280,height=720,format=BGR ! appsink drop=1'
+			return 'rtspsrc location='+input()+' latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! video/x-raw,format=BGR ! appsink drop=1'
+		case 'prompt_rtsp_h265':
+			print("Please enter the full link")
+			return 'rtspsrc location='+input()+' latency=100 ! queue ! rtph265depay ! avdec_h265 ! videoconvert ! video/x-raw,format=BGR ! appsink drop=1'
 		case _:
 			return 'Option invalide'
 

App/Demo_squelet/ia_squelet_x2.py

@@ -54,9 +54,12 @@ def switch_case(argument):
 		case 'prompt_ip_webcam':
 			print("Please enter the device's IP address")
 			return 'rtspsrc location=rtsp://'+input()+':8080/h264_ulaw.sdp latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! videoscale ! video/x-raw,width=1280,height=720,format=BGR ! appsink drop=1'
-		case 'prompt_rtsp':
+		case 'prompt_rtsp_h264':
 			print("Please enter the full link")
-			return  'rtspsrc location='+input()+' latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! videoscale ! video/x-raw,width=1280,height=720,format=BGR ! appsink drop=1'
+			return 'rtspsrc location='+input()+' latency=100 ! queue ! rtph264depay ! avdec_h264 ! videoconvert ! video/x-raw,format=BGR ! appsink drop=1'
+		case 'prompt_rtsp_h265':
+			print("Please enter the full link")
+			return 'rtspsrc location='+input()+' latency=100 ! queue ! rtph265depay ! avdec_h265 ! videoconvert ! video/x-raw,format=BGR ! appsink drop=1'
 		case _:
 			return 'Option invalide' 
 

execute.sh

@@ -13,7 +13,7 @@ execute_docker () {
 			;;
 		esac
 	fi
-	echo $'What would you like to do?\n1. Execute with webcam\n2. Execute with phone app "IP Webcam"\n3. Execute with nano2 5G box\n4. Execute with another RTSP source'
+	echo $'What would you like to do?\n1. Execute with webcam\n2. Execute with phone app "IP Webcam"\n3. Execute with nano2 5G box\n4. Execute with another H264 RTSP source\n5. Execute with another H265 RTSP source'
 	read choice
 	case $choice in
 	1)
@@ -30,7 +30,11 @@ execute_docker () {
 		;;
 	4)
 		echo 'The container is starting... (please wait for it to prompt for link)'
-		docker run -i --rm --network host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix $image_name:$installed_version python3 Demo_squelet/ia_squelet.py --source prompt_rtsp
+		docker run -i --rm --network host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix $image_name:$installed_version python3 Demo_squelet/ia_squelet.py --source prompt_rtsp_h264
+		;;
+	5)
+		echo 'The container is starting... (please wait for it to prompt for link)'
+		docker run -i --rm --network host --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix:/tmp/.X11-unix $image_name:$installed_version python3 Demo_squelet/ia_squelet.py --source prompt_rtsp_h265
 		;;
 	*)
 		echo 'Wrong choice'