Upload 11 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+2024-05-18_14-46-07_4910.png filter=lfs diff=lfs merge=lfs -text
+2024-05-18_14-57-09_5235.png filter=lfs diff=lfs merge=lfs -text

PS1/yolov8n.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:31e20dde3def09e2cf938c7be6fe23d9150bbbe503982af13345706515f2ef95
+size 6534387

PS2/DQN_MODELS/DQN_MODEL1.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9b7923d58703d779fdaee6b986dfbcad66332f42402536951864f97b82f067a
+size 65427

PS2/single-intersection-gen.rou.xml

@@ -0,0 +1,61 @@
+<routes>
+                <route id="route_ns" edges="n_t t_s"/>
+                <route id="route_nw" edges="n_t t_w"/>
+                <route id="route_ne" edges="n_t t_e"/>
+                <route id="route_we" edges="w_t t_e"/>
+                <route id="route_wn" edges="w_t t_n"/>
+                <route id="route_ws" edges="w_t t_s"/>
+                <route id="route_ew" edges="e_t t_w"/>
+                <route id="route_en" edges="e_t t_n"/>
+                <route id="route_es" edges="e_t t_s"/>
+                <route id="route_sn" edges="s_t t_n"/>
+                <route id="route_se" edges="s_t t_e"/>
+                <route id="route_sw" edges="s_t t_w"/><flow id="flow_ns_0" route="route_ns" begin="0" end="100000" vehsPerHour="100" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_nw_0" route="route_nw" begin="0" end="100000" vehsPerHour="100" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ne_0" route="route_ne" begin="0" end="100000" vehsPerHour="200" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sw_0" route="route_sw" begin="0" end="100000" vehsPerHour="200" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sn_0" route="route_sn" begin="0" end="100000" vehsPerHour="100" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_se_0" route="route_se" begin="0" end="100000" vehsPerHour="100" departSp100000d="max" departPos="base" departLane="best"/>
+
+    <flow id="flow_en_0" route="route_en" begin="0" end="100000" vehsPerHour="300" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ew_0" route="route_ew" begin="0" end="100000" vehsPerHour="300" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_es_0" route="route_es" begin="0" end="100000" vehsPerHour="600" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_wn_0" route="route_wn" begin="0" end="100000" vehsPerHour="600" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_we_0" route="route_we" begin="0" end="100000" vehsPerHour="300" departSp100000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ws_0" route="route_ws" begin="0" end="100000" vehsPerHour="300" departSp100000d="max" departPos="base" departLane="best"/><flow id="flow_ns_1" route="route_ns" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_nw_1" route="route_nw" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ne_1" route="route_ne" begin="100000" end="200000" vehsPerHour="400" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sw_1" route="route_sw" begin="100000" end="200000" vehsPerHour="400" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sn_1" route="route_sn" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_se_1" route="route_se" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+
+    <flow id="flow_en_1" route="route_en" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ew_1" route="route_ew" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_es_1" route="route_es" begin="100000" end="200000" vehsPerHour="400" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_wn_1" route="route_wn" begin="100000" end="200000" vehsPerHour="400" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_we_1" route="route_we" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ws_1" route="route_ws" begin="100000" end="200000" vehsPerHour="200" departSp200000d="max" departPos="base" departLane="best"/><flow id="flow_ns_2" route="route_ns" begin="200000" end="300000" vehsPerHour="100" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_nw_2" route="route_nw" begin="200000" end="300000" vehsPerHour="100" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ne_2" route="route_ne" begin="200000" end="300000" vehsPerHour="200" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sw_2" route="route_sw" begin="200000" end="300000" vehsPerHour="200" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sn_2" route="route_sn" begin="200000" end="300000" vehsPerHour="100" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_se_2" route="route_se" begin="200000" end="300000" vehsPerHour="100" departSp300000d="max" departPos="base" departLane="best"/>
+
+    <flow id="flow_en_2" route="route_en" begin="200000" end="300000" vehsPerHour="300" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ew_2" route="route_ew" begin="200000" end="300000" vehsPerHour="300" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_es_2" route="route_es" begin="200000" end="300000" vehsPerHour="600" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_wn_2" route="route_wn" begin="200000" end="300000" vehsPerHour="600" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_we_2" route="route_we" begin="200000" end="300000" vehsPerHour="300" departSp300000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ws_2" route="route_ws" begin="200000" end="300000" vehsPerHour="300" departSp300000d="max" departPos="base" departLane="best"/><flow id="flow_ns_3" route="route_ns" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_nw_3" route="route_nw" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ne_3" route="route_ne" begin="300000" end="400000" vehsPerHour="400" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sw_3" route="route_sw" begin="300000" end="400000" vehsPerHour="400" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_sn_3" route="route_sn" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_se_3" route="route_se" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+
+    <flow id="flow_en_3" route="route_en" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ew_3" route="route_ew" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_es_3" route="route_es" begin="300000" end="400000" vehsPerHour="400" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_wn_3" route="route_wn" begin="300000" end="400000" vehsPerHour="400" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_we_3" route="route_we" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/>
+    <flow id="flow_ws_3" route="route_ws" begin="300000" end="400000" vehsPerHour="200" departSp400000d="max" departPos="base" departLane="best"/></routes>

PS2/single-intersection.net.xml

@@ -0,0 +1,186 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<!-- generated on sáb 26 jan 2019 20:40:10 -02 by Netedit Version 0.32.0
+<?xml version="1.0" encoding="UTF-8"?>
+
+<configuration xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="http://sumo.dlr.de/xsd/netconvertConfiguration.xsd">
+
+    <input>
+        <sumo-net-file value="single-intersection.net.xml"/>
+    </input>
+
+    <output>
+        <output-file value="single-intersection.net.xml"/>
+    </output>
+
+    <processing>
+        <no-turnarounds value="true"/>
+        <offset.disable-normalization value="true"/>
+        <lefthand value="false"/>
+        <junctions.corner-detail value="0"/>
+        <rectangular-lane-cut value="false"/>
+        <walkingareas value="false"/>
+    </processing>
+
+</configuration>
+-->
+
+<net version="0.27" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:noNamespaceSchemaLocation="http://sumo.dlr.de/xsd/net_file.xsd">
+
+    <location netOffset="150.00,150.00" convBoundary="-0.07,0.00,300.00,300.00" origBoundary="-150.00,-150.00,150.00,150.00" projParameter="!"/>
+
+    <edge id=":e_0" function="internal">
+        <lane id=":e_0_0" index="0" speed="13.90" length="4.82" shape="300.00,148.35 301.24,149.17 301.65,150.00 301.24,150.82 300.00,151.65"/>
+    </edge>
+    <edge id=":n_0" function="internal">
+        <lane id=":n_0_0" index="0" speed="13.90" length="4.82" shape="151.65,300.00 150.83,301.24 150.00,301.65 149.18,301.24 148.35,300.00"/>
+    </edge>
+    <edge id=":s_0" function="internal">
+        <lane id=":s_0_0" index="0" speed="13.90" length="4.82" shape="148.35,0.00 149.17,-1.24 150.00,-1.65 150.82,-1.24 151.65,0.00"/>
+    </edge>
+    <edge id=":t_0" function="internal">
+        <lane id=":t_0_0" index="0" speed="13.90" length="5.00" shape="145.05,158.05 144.86,156.69 144.28,155.73 143.31,155.14 141.95,154.95"/>
+    </edge>
+    <edge id=":t_1" function="internal">
+        <lane id=":t_1_0" index="0" speed="13.90" length="16.10" shape="145.05,158.05 145.05,141.95"/>
+    </edge>
+    <edge id=":t_2" function="internal">
+        <lane id=":t_2_0" index="0" speed="13.90" length="15.64" shape="148.35,158.05 148.96,153.81 150.77,150.77 153.81,148.96 158.05,148.35"/>
+    </edge>
+    <edge id=":t_3" function="internal">
+        <lane id=":t_3_0" index="0" speed="13.90" length="5.00" shape="158.05,154.95 156.69,155.14 155.72,155.72 155.14,156.69 154.95,158.05"/>
+    </edge>
+    <edge id=":t_4" function="internal">
+        <lane id=":t_4_0" index="0" speed="13.90" length="16.10" shape="158.05,154.95 141.95,154.95"/>
+    </edge>
+    <edge id=":t_5" function="internal">
+        <lane id=":t_5_0" index="0" speed="13.90" length="15.64" shape="158.05,151.65 153.81,151.04 150.77,149.23 148.96,146.19 148.35,141.95"/>
+    </edge>
+    <edge id=":t_6" function="internal">
+        <lane id=":t_6_0" index="0" speed="13.90" length="5.00" shape="154.95,141.95 155.14,143.31 155.72,144.28 156.69,144.86 158.05,145.05"/>
+    </edge>
+    <edge id=":t_7" function="internal">
+        <lane id=":t_7_0" index="0" speed="13.90" length="16.10" shape="154.95,141.95 154.95,158.05"/>
+    </edge>
+    <edge id=":t_8" function="internal">
+        <lane id=":t_8_0" index="0" speed="13.90" length="15.64" shape="151.65,141.95 151.04,146.19 149.23,149.22 146.19,151.04 141.95,151.65"/>
+    </edge>
+    <edge id=":t_9" function="internal">
+        <lane id=":t_9_0" index="0" speed="13.90" length="5.00" shape="141.95,145.05 143.31,144.86 144.28,144.28 144.86,143.31 145.05,141.95"/>
+    </edge>
+    <edge id=":t_10" function="internal">
+        <lane id=":t_10_0" index="0" speed="13.90" length="16.10" shape="141.95,145.05 158.05,145.05"/>
+    </edge>
+    <edge id=":t_11" function="internal">
+        <lane id=":t_11_0" index="0" speed="13.90" length="15.64" shape="141.95,148.35 146.19,148.96 149.23,150.77 151.04,153.81 151.65,158.05"/>
+    </edge>
+    <edge id=":w_0" function="internal">
+        <lane id=":w_0_0" index="0" speed="13.90" length="4.82" shape="-0.07,151.65 -1.29,150.80 -1.69,149.97 -1.26,149.15 0.00,148.35"/>
+    </edge>
+
+    <edge id="e_t" from="e" to="t" priority="-1">
+        <lane id="e_t_0" index="0" speed="13.90" length="141.95" shape="300.00,154.95 158.05,154.95"/>
+        <lane id="e_t_1" index="1" speed="13.90" length="141.95" shape="300.00,151.65 158.05,151.65"/>
+    </edge>
+    <edge id="n_t" from="n" to="t" priority="-1">
+        <lane id="n_t_0" index="0" speed="13.90" length="141.95" shape="145.05,300.00 145.05,158.05"/>
+        <lane id="n_t_1" index="1" speed="13.90" length="141.95" shape="148.35,300.00 148.35,158.05"/>
+    </edge>
+    <edge id="s_t" from="s" to="t" priority="-1">
+        <lane id="s_t_0" index="0" speed="13.90" length="141.95" shape="154.95,0.00 154.95,141.95"/>
+        <lane id="s_t_1" index="1" speed="13.90" length="141.95" shape="151.65,0.00 151.65,141.95"/>
+    </edge>
+    <edge id="t_e" from="t" to="e" priority="-1">
+        <lane id="t_e_0" index="0" speed="13.90" length="141.95" shape="158.05,145.05 300.00,145.05"/>
+        <lane id="t_e_1" index="1" speed="13.90" length="141.95" shape="158.05,148.35 300.00,148.35"/>
+    </edge>
+    <edge id="t_n" from="t" to="n" priority="-1">
+        <lane id="t_n_0" index="0" speed="13.90" length="141.95" shape="154.95,158.05 154.95,300.00"/>
+        <lane id="t_n_1" index="1" speed="13.90" length="141.95" shape="151.65,158.05 151.65,300.00"/>
+    </edge>
+    <edge id="t_s" from="t" to="s" priority="-1">
+        <lane id="t_s_0" index="0" speed="13.90" length="141.95" shape="145.05,141.95 145.05,0.00"/>
+        <lane id="t_s_1" index="1" speed="13.90" length="141.95" shape="148.35,141.95 148.35,0.00"/>
+    </edge>
+    <edge id="t_w" from="t" to="w" priority="-1" spreadType="center" shape="150.00,153.30 -0.07,153.30">
+        <lane id="t_w_0" index="0" speed="13.90" length="142.02" shape="141.95,154.95 -0.07,154.95"/>
+        <lane id="t_w_1" index="1" speed="13.90" length="142.02" shape="141.95,151.65 -0.07,151.65"/>
+    </edge>
+    <edge id="w_t" from="w" to="t" priority="-1">
+        <lane id="w_t_0" index="0" speed="13.90" length="141.95" shape="0.00,145.05 141.95,145.05"/>
+        <lane id="w_t_1" index="1" speed="13.90" length="141.95" shape="0.00,148.35 141.95,148.35"/>
+    </edge>
+
+    <tlLogic id="t" type="static" programID="0" offset="0">
+        <phase duration="33" state="GGrrrrGGrrrr"/>
+        <phase duration="2" state="yyrrrryyrrrr"/>
+        <phase duration="6" state="rrGrrrrrGrrr"/>
+        <phase duration="2" state="rryrrrrryrrr"/>
+        <phase duration="33" state="rrrGGrrrrGGr"/>
+        <phase duration="2" state="rrryyrrrryyr"/>
+        <phase duration="6" state="rrrrrGrrrrrG"/>
+        <phase duration="2" state="rrrrryrrrrry"/>
+    </tlLogic>
+
+    <junction id="e" type="priority" x="300.00" y="150.00" incLanes="t_e_0 t_e_1" intLanes=":e_0_0" shape="300.00,150.05 300.00,143.45 300.00,149.95">
+        <request index="0" response="0" foes="0" cont="0"/>
+    </junction>
+    <junction id="n" type="priority" x="150.00" y="300.00" incLanes="t_n_0 t_n_1" intLanes=":n_0_0" shape="149.95,300.00 156.55,300.00 150.05,300.00">
+        <request index="0" response="0" foes="0" cont="0"/>
+    </junction>
+    <junction id="s" type="priority" x="150.00" y="0.00" incLanes="t_s_0 t_s_1" intLanes=":s_0_0" shape="150.05,0.00 143.45,0.00 149.95,0.00">
+        <request index="0" response="0" foes="0" cont="0"/>
+    </junction>
+    <junction id="t" type="traffic_light" x="150.00" y="150.00" incLanes="n_t_0 n_t_1 e_t_0 e_t_1 s_t_0 s_t_1 w_t_0 w_t_1" intLanes=":t_0_0 :t_1_0 :t_2_0 :t_3_0 :t_4_0 :t_5_0 :t_6_0 :t_7_0 :t_8_0 :t_9_0 :t_10_0 :t_11_0" shape="143.45,158.05 156.55,158.05 158.05,156.55 158.05,143.45 156.55,141.95 143.45,141.95 141.95,143.45 141.95,156.55">
+        <request index="0" response="000000000000" foes="000100010000" cont="0"/>
+        <request index="1" response="000000000000" foes="111100110000" cont="0"/>
+        <request index="2" response="000011000000" foes="110011110000" cont="0"/>
+        <request index="3" response="000010000000" foes="100010000000" cont="0"/>
+        <request index="4" response="000110000111" foes="100110000111" cont="0"/>
+        <request index="5" response="011110000110" foes="011110000110" cont="0"/>
+        <request index="6" response="000000000000" foes="010000000100" cont="0"/>
+        <request index="7" response="000000000000" foes="110000111100" cont="0"/>
+        <request index="8" response="000000000011" foes="110000110011" cont="0"/>
+        <request index="9" response="000000000010" foes="000000100010" cont="0"/>
+        <request index="10" response="000111000110" foes="000111100110" cont="0"/>
+        <request index="11" response="000110011110" foes="000110011110" cont="0"/>
+    </junction>
+    <junction id="w" type="priority" x="0.00" y="150.00" incLanes="t_w_0 t_w_1" intLanes=":w_0_0" shape="0.00,149.95 0.00,156.55 0.00,150.05">
+        <request index="0" response="0" foes="0" cont="0"/>
+    </junction>
+
+    <connection from="e_t" to="t_n" fromLane="0" toLane="0" via=":t_3_0" tl="t" linkIndex="3" dir="r" state="o"/>
+    <connection from="e_t" to="t_w" fromLane="0" toLane="0" via=":t_4_0" tl="t" linkIndex="4" dir="s" state="o"/>
+    <connection from="e_t" to="t_s" fromLane="1" toLane="1" via=":t_5_0" tl="t" linkIndex="5" dir="l" state="o"/>
+    <connection from="n_t" to="t_w" fromLane="0" toLane="0" via=":t_0_0" tl="t" linkIndex="0" dir="r" state="o"/>
+    <connection from="n_t" to="t_s" fromLane="0" toLane="0" via=":t_1_0" tl="t" linkIndex="1" dir="s" state="o"/>
+    <connection from="n_t" to="t_e" fromLane="1" toLane="1" via=":t_2_0" tl="t" linkIndex="2" dir="l" state="o"/>
+    <connection from="s_t" to="t_e" fromLane="0" toLane="0" via=":t_6_0" tl="t" linkIndex="6" dir="r" state="o"/>
+    <connection from="s_t" to="t_n" fromLane="0" toLane="0" via=":t_7_0" tl="t" linkIndex="7" dir="s" state="o"/>
+    <connection from="s_t" to="t_w" fromLane="1" toLane="1" via=":t_8_0" tl="t" linkIndex="8" dir="l" state="o"/>
+    <connection from="t_e" to="e_t" fromLane="1" toLane="1" via=":e_0_0" dir="t" state="M"/>
+    <connection from="t_n" to="n_t" fromLane="1" toLane="1" via=":n_0_0" dir="t" state="M"/>
+    <connection from="t_s" to="s_t" fromLane="1" toLane="1" via=":s_0_0" dir="t" state="M"/>
+    <connection from="t_w" to="w_t" fromLane="1" toLane="1" via=":w_0_0" dir="t" state="M"/>
+    <connection from="w_t" to="t_s" fromLane="0" toLane="0" via=":t_9_0" tl="t" linkIndex="9" dir="r" state="o"/>
+    <connection from="w_t" to="t_e" fromLane="0" toLane="0" via=":t_10_0" tl="t" linkIndex="10" dir="s" state="o"/>
+    <connection from="w_t" to="t_n" fromLane="1" toLane="1" via=":t_11_0" tl="t" linkIndex="11" dir="l" state="o"/>
+
+    <connection from=":e_0" to="e_t" fromLane="0" toLane="1" dir="t" state="M"/>
+    <connection from=":n_0" to="n_t" fromLane="0" toLane="1" dir="t" state="M"/>
+    <connection from=":s_0" to="s_t" fromLane="0" toLane="1" dir="t" state="M"/>
+    <connection from=":t_0" to="t_w" fromLane="0" toLane="0" dir="r" state="M"/>
+    <connection from=":t_1" to="t_s" fromLane="0" toLane="0" dir="s" state="M"/>
+    <connection from=":t_2" to="t_e" fromLane="0" toLane="1" dir="l" state="M"/>
+    <connection from=":t_3" to="t_n" fromLane="0" toLane="0" dir="r" state="M"/>
+    <connection from=":t_4" to="t_w" fromLane="0" toLane="0" dir="s" state="M"/>
+    <connection from=":t_5" to="t_s" fromLane="0" toLane="1" dir="l" state="M"/>
+    <connection from=":t_6" to="t_e" fromLane="0" toLane="0" dir="r" state="M"/>
+    <connection from=":t_7" to="t_n" fromLane="0" toLane="0" dir="s" state="M"/>
+    <connection from=":t_8" to="t_w" fromLane="0" toLane="1" dir="l" state="M"/>
+    <connection from=":t_9" to="t_s" fromLane="0" toLane="0" dir="r" state="M"/>
+    <connection from=":t_10" to="t_e" fromLane="0" toLane="0" dir="s" state="M"/>
+    <connection from=":t_11" to="t_n" fromLane="0" toLane="1" dir="l" state="M"/>
+    <connection from=":w_0" to="w_t" fromLane="0" toLane="1" dir="t" state="M"/>
+
+</net>

PS3/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b5b6ab79f1ff5b0e28a2715ac0bee9eb93c13aaa5214a2bfde2ab617c70a74e8
+size 6780515

PS3/yolov8n.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7eb70257593da06f682a3ddda54a9d260d4fc514f645237f5ca74b08f8da61a6
+size 2

app.py

@@ -0,0 +1,419 @@
+from ultralytics import YOLO
+import base64
+import cv2
+import io
+import numpy as np
+from ultralytics.utils.plotting import Annotator
+import streamlit as st
+from streamlit_image_coordinates import streamlit_image_coordinates
+import pandas as pd
+import ollama
+import bs4
+from langchain.text_splitter import RecursiveCharacterTextSplitter
+from langchain_community.document_loaders import WebBaseLoader
+from langchain_community.document_loaders import CSVLoader
+from langchain_community.vectorstores import Chroma
+from langchain_community.embeddings import OllamaEmbeddings
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import RunnablePassthrough
+
+def set_background(image_file1,image_file2):
+    
+    with open(image_file1, "rb") as f:
+        img_data1 = f.read()
+    b64_encoded1 = base64.b64encode(img_data1).decode()
+    with open(image_file2, "rb") as f:
+        img_data2 = f.read()
+    b64_encoded2 = base64.b64encode(img_data2).decode()
+    style = f"""
+        <style>
+        .stApp{{
+            background-image: url(data:image/png;base64,{b64_encoded1});
+            background-size: cover;
+            
+        }}
+        .st-emotion-cache-6qob1r{{
+            background-image: url(data:image/png;base64,{b64_encoded2});
+            background-size: cover;
+            border: 5px solid rgb(14, 17, 23);
+            
+        }}
+        </style>
+    """
+    st.markdown(style, unsafe_allow_html=True)
+
+set_background('pngtree-city-map-navigation-interface-picture-image_1833642.png','2024-05-18_14-57-09_5235.png')
+
+st.title("Traffic Flow and Optimization Toolkit")
+
+sb = st.sidebar # defining the sidebar
+
+sb.markdown("🛰️ **Navigation**")
+page_names = ["PS1", "PS2", "PS3","Chat with Results"]
+page = sb.radio("", page_names, index=0)
+st.session_state['n'] = sb.slider("Number of ROIs",1,5)
+
+if page == 'PS1':
+    uploaded_file = st.file_uploader("Choose a video...", type=["mp4", "mpeg"])
+    if uploaded_file is not None:
+        g = io.BytesIO(uploaded_file.read()) 
+        temporary_location = "temp_PS1.mp4"
+
+        with open(temporary_location, 'wb') as out:  
+            out.write(g.read())  
+        out.close()
+        model = YOLO('PS1\yolov8n.pt')
+        if 'roi_list1' not in st.session_state:
+            st.session_state['roi_list1'] = []
+        if "all_rois1" not in st.session_state:    
+            st.session_state['all_rois1'] = []
+        classes = model.names
+
+        done_1 = st.button('Selection Done')
+
+        while len(st.session_state["all_rois1"]) < st.session_state['n']:
+            cap = cv2.VideoCapture('temp_PS1.mp4')
+            while not done_1:
+                ret,frame=cap.read()
+                cv2.putText(frame,'SELECT ROI',(100,100),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),4)
+                if not ret:
+                    st.write('ROI selection has concluded')
+                    break
+                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                value = streamlit_image_coordinates(frame,key='numpy',width=750)
+                st.session_state["roi_list1"].append([int(value['x']*2.55),int(value['y']*2.55)])
+                st.write(st.session_state["roi_list1"])
+                if cv2.waitKey(0)&0xFF==27:
+                    break
+            cap.release()
+            st.session_state["all_rois1"].append(st.session_state["roi_list1"])
+            st.session_state["roi_list1"] = []
+            done_1 = False
+
+        st.write('ROI indices: ',st.session_state["all_rois1"][0])
+
+
+
+        cap = cv2.VideoCapture('temp_PS1.MP4')
+        st.write("Detection started")
+        st.session_state['fps'] = cap.get(cv2.CAP_PROP_FPS)
+        st.write(f"FPS OF VIDEO: {st.session_state['fps']}")
+        avg_list = []
+        count = 0
+        frame_placeholder = st.empty()
+        st.session_state["data1"] = {}
+        for i in range(len(st.session_state["all_rois1"])):
+            st.session_state["data1"][f"ROI{i}"] = []
+        while cap.isOpened():
+            ret,frame=cap.read()
+            if not ret:
+                break
+            count += 1
+            if count % 3 != 0:
+                continue
+            k = 0
+            for roi_list_here1 in st.session_state["all_rois1"]:
+                max = [0,0]
+                min = [10000,10000]
+                roi_list_here = roi_list_here1[1:]
+                for i in range(len(roi_list_here)):
+                    if roi_list_here[i][0] > max[0]:
+                        max[0] = roi_list_here[i][0]
+                    if roi_list_here[i][1] > max[1]:
+                        max[1] = roi_list_here[i][1]
+                    if roi_list_here[i][0] < min[0]:
+                        min[0] = roi_list_here[i][0]
+                    if roi_list_here[i][1] < min[1]:
+                        min[1] = roi_list_here[i][1]
+                frame_cropped = frame[min[1]:max[1],min[0]:max[0]]
+                roi_corners = np.array([roi_list_here],dtype=np.int32)
+                mask = np.zeros(frame.shape,dtype=np.uint8)
+                mask.fill(255)
+                channel_count = frame.shape[2]
+                ignore_mask_color = (255,)*channel_count
+                cv2.fillPoly(mask,roi_corners,0)
+                mask_cropped = mask[min[1]:max[1],min[0]:max[0]]
+                roi = cv2.bitwise_or(frame_cropped,mask_cropped)
+
+                #roi = frame[roi_list_here[0][1]:roi_list_here[1][1],roi_list_here[0][0]:roi_list_here[1][0]]
+                number = []
+                results = model.predict(roi)
+                for r in results:
+                    boxes = r.boxes
+                    counter = 0
+                    for box in boxes:
+                        counter += 1
+                        name = classes[box.cls.numpy()[0]]
+                        conf = str(round(box.conf.numpy()[0],2))
+                        text = name+""+conf
+                        bbox = box.xyxy[0].numpy()
+                        cv2.rectangle(frame,(int(bbox[0])+min[0],int(bbox[1])+min[1]),(int(bbox[2])+min[0],int(bbox[3])+min[1]),(0,255,0),2)
+                        cv2.putText(frame,text,(int(bbox[0])+min[0],int(bbox[1])+min[1]-5),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),2)
+                    number.append(counter)
+                avg = sum(number)/len(number)
+                stats = str(round(avg,2))
+                if count%10 == 0:
+                    st.session_state["data1"][f"ROI{k}"].append(avg)
+                    k+=1
+                cv2.putText(frame,stats,(min[0],min[1]),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,0),4)
+                cv2.polylines(frame,roi_corners,True,(255,0,0),2)
+            cv2.putText(frame,'The average number of vehicles in the Regions of Interest',(100,100),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),4)
+            frame_placeholder.image(frame,channels='BGR')
+        cap.release()
+        df = pd.DataFrame(st.session_state["data1"])
+        df.to_csv('PS1.csv', sep='\t', encoding='utf-8')
+    else:
+        st.error('PLEASE UPLOAD AN IMAGE OF THE FORMAT JPG,JPEG OR PNG', icon="🚨")
+
+elif page == "PS3":
+    uploaded_file1 = st.file_uploader("Choose a video...", type=["mp4", "mpeg"])
+    if uploaded_file1 is not None:
+        g = io.BytesIO(uploaded_file1.read()) 
+        temporary_location = "temp_PS2.mp4"
+
+        with open(temporary_location, 'wb') as out:  
+            out.write(g.read())  
+        out.close()
+        model1 = YOLO("PS1\yolov8n.pt")
+        model2 = YOLO(r"PS3\best.pt")
+        if 'roi_list2' not in st.session_state:
+            st.session_state['roi_list2'] = []
+        if "all_rois2" not in st.session_state:    
+            st.session_state['all_rois2'] = []
+        classes = model1.names
+
+        done_2 = st.button('Selection Done')
+
+        while len(st.session_state["all_rois2"]) < st.session_state['n']:
+            cap = cv2.VideoCapture('temp_PS2.mp4')
+            while not done_2:
+                ret,frame=cap.read()
+                cv2.putText(frame,'SELECT ROI',(100,100),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),4)
+                if not ret:
+                    st.write('ROI selection has concluded')
+                    break
+                frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+                value = streamlit_image_coordinates(frame,key='numpy',width=750)
+                st.session_state["roi_list2"].append([int(value['x']*2.5),int(value['y']*2.5)])
+                st.write(st.session_state["roi_list2"])
+                if cv2.waitKey(0)&0xFF==27:
+                    break
+            cap.release()
+            st.session_state["all_rois2"].append(st.session_state["roi_list2"])
+            st.session_state["roi_list2"] = []
+            done_2 = False
+
+        st.write('ROI indices: ',st.session_state["all_rois2"][0])
+
+
+
+        cap = cv2.VideoCapture('temp_PS2.MP4')
+        st.write("Detection started")
+        avg_list = []
+        count = 0
+        frame_placeholder = st.empty()
+        st.session_state.data = {}
+        for i in range(len(st.session_state["all_rois2"])):
+            st.session_state["data"][f"ROI{i}"] = []
+        for i in range(len(st.session_state['all_rois2'])):
+            st.session_state.data[f"ROI{i}"] = []
+        while cap.isOpened():
+            ret,frame=cap.read()
+            if not ret:
+                break
+            count += 1
+            if count % 3 != 0:
+                continue
+            # rois = []
+            k = 0
+            for roi_list_here1 in st.session_state["all_rois2"]:
+                max = [0,0]
+                min = [10000,10000]
+                roi_list_here = roi_list_here1[1:]
+                for i in range(len(roi_list_here)-1):
+                    if roi_list_here[i][0] > max[0]:
+                        max[0] = roi_list_here[i][0]
+                    if roi_list_here[i][1] > max[1]:
+                        max[1] = roi_list_here[i][1]
+                    if roi_list_here[i][0] < min[0]:
+                        min[0] = roi_list_here[i][0]
+                    if roi_list_here[i][1] < min[1]:
+                        min[1] = roi_list_here[i][1]
+                frame_cropped = frame[min[1]:max[1],min[0]:max[0]]
+                roi_corners = np.array([roi_list_here],dtype=np.int32)
+                mask = np.zeros(frame.shape,dtype=np.uint8)
+                mask.fill(255)
+                channel_count = frame.shape[2]
+                ignore_mask_color = (255,)*channel_count
+                cv2.fillPoly(mask,roi_corners,0)
+                mask_cropped = mask[min[1]:max[1],min[0]:max[0]]
+                roi = cv2.bitwise_or(frame_cropped,mask_cropped)
+
+                #roi = frame[roi_list_here[0][1]:roi_list_here[1][1],roi_list_here[0][0]:roi_list_here[1][0]]
+                number = []
+                results = model1.predict(roi)
+                results_pothole = model2.predict(source=frame)
+                for r in results:
+                    boxes = r.boxes
+                    counter = 0
+                    for box in boxes:
+                        counter += 1
+                        name = classes[box.cls.numpy()[0]]
+                        conf = str(round(box.conf.numpy()[0],2))
+                        text = name+conf
+                        bbox = box.xyxy[0].numpy()
+                        cv2.rectangle(frame,(int(bbox[0])+min[0],int(bbox[1])+min[1]),(int(bbox[2])+min[0],int(bbox[3])+min[1]),(0,255,0),2)
+                        cv2.putText(frame,text,(int(bbox[0])+min[0],int(bbox[1])+min[1]-5),cv2.FONT_HERSHEY_SIMPLEX, 0.4,(0,0,255),2)
+                    number.append(counter)
+                for r in results_pothole:
+                    masks = r.masks
+                    boxes = r.boxes.cpu().numpy()
+                    xyxys = boxes.xyxy
+                    confs = boxes.conf
+                    if masks is not None:
+                        shapes = np.ones_like(frame)
+                        for mask,conf,xyxy in zip(masks,confs,xyxys):
+                            polygon = mask.xy[0]
+                            if conf >= 0.49 and len(polygon)>=3:
+                                cv2.fillPoly(shapes,pts=np.int32([polygon]),color=(0,0,255,0.5))
+                                frame = cv2.addWeighted(frame,0.7,shapes,0.3,gamma=0)
+                                cv2.rectangle(frame,(int(xyxy[0]),int(xyxy[1])),(int(xyxy[2]),int(xyxy[3])),(0,0,255),2)
+                                cv2.putText(frame,'Pothole '+str(conf),(int(xyxy[0]),int(xyxy[1])-5),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),2)
+
+                avg = sum(number)/len(number)
+                stats = str(round(avg,2))
+                if count % 10 == 0:
+                    st.session_state.data[f"ROI{k}"].append(avg)
+                    k+=1
+                cv2.putText(frame,stats,(min[0],min[1]),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,0),4)
+                cv2.polylines(frame,roi_corners,True,(255,0,0),2)
+                if counter >= 5:
+                    cv2.putText(frame,'!!CONGESTION MORE THAN '+str(counter)+' Objects',(min[0]+20,min[1]+20),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,0),4)
+                cv2.polylines(frame,roi_corners,True,(255,0,0),2)
+                cv2.putText(frame,'Objects in the Regions of Interest',(100,100),cv2.FONT_HERSHEY_SIMPLEX, 1,(0,0,255),4)
+            frame_placeholder.image(frame,channels='BGR')
+        cap.release()
+
+        df = pd.DataFrame(st.session_state.data)
+        df.to_csv('PS2.csv', sep='\t', encoding='utf-8')
+
+    else:
+        st.error('PLEASE UPLOAD AN IMAGE OF THE FORMAT JPG,JPEG OR PNG', icon="🚨")
+
+elif page == "PS2":
+    st.header("CLICK ON RUN SCRIPT TO START A TRAFFIC SIMULATION")
+    script = st.button("RUN SCRIPT")
+    st.session_state.con = -1
+    if script:
+        st.session_state.con += 1
+        import gymnasium as gym
+        import sumo_rl
+        import os
+        from stable_baselines3 import DQN
+        from stable_baselines3.common.vec_env import DummyVecEnv
+        from stable_baselines3.common.evaluation import evaluate_policy
+        from sumo_rl import SumoEnvironment
+        env = gym.make('sumo-rl-v0',
+                net_file='PS2\single-intersection.net.xml',
+                route_file='PS2\single-intersection-gen.rou.xml',
+                out_csv_name='output',
+                use_gui=True,
+                single_agent=True,    
+                num_seconds=10000)
+        model1 = DQN.load('PS2\DQN_MODELS\DQN_MODEL3.zip',env=env)
+        one,two = evaluate_policy(model1,env = env,n_eval_episodes=5,render=True)
+        st.write("Evaluation Results: ",one,two)
+        import matplotlib.pyplot as plt
+        def eval_plot(path,metric,path_compare = None):
+            data = pd.read_csv(path)
+            if path_compare is not None:
+                data1 = pd.read_csv(path_compare)
+            x = []
+            for i in range(0,len(data)):
+                x.append(i)
+
+            y = data[metric]
+            y_1 = pd.to_numeric(y)
+            y_arr = np.array(y_1)
+            if path_compare is not None:
+                y2 = data1[metric]
+                y_2 = pd.to_numeric(y2)
+                y_arr2 = np.array(y_2)
+
+            x_arr = np.array(x)
+
+            fig = plt.figure()
+            ax1 = fig.add_subplot(2, 1, 1)
+            ax1.set_title(metric)
+            if path_compare is not None:
+                ax2 = fig.add_subplot(2, 1, 2,sharey=ax1)
+                ax2.set_title('compare '+metric)
+
+            ax1.plot(x_arr,y_arr)
+            
+            if path_compare  is not None:
+                ax2.plot(x_arr,y_arr2)
+            
+            return fig
+        for i in range(1,2):
+            st.pyplot(eval_plot(f'output_conn{st.session_state.con}_ep{i}.csv','system_mean_waiting_time'))
+            st.pyplot(eval_plot(f'output_conn{st.session_state.con}_ep{i}.csv','agents_total_accumulated_waiting_time'))  
+
+elif page == "Chat with Results":
+    st.title('Chat with the Results')
+    st.write("Please upload the relevant CSV data to get started")
+    reload = st.button('Reload')
+    if 'isran' not in st.session_state or reload == True:
+        st.session_state['isran'] = False
+
+
+    uploaded_file = st.file_uploader('Choose your .csv file', type=["csv"])
+    if uploaded_file is not None and st.session_state['isran'] == False:
+        with open("temp.csv", "wb") as f:
+            f.write(uploaded_file.getvalue())
+        loader = CSVLoader('temp.csv')
+        docs = loader.load()
+        text_splitter = RecursiveCharacterTextSplitter(chunk_size = 1000, chunk_overlap = 200)
+        splits = text_splitter.split_documents(docs)
+        
+        embeddings = OllamaEmbeddings(model='mistral')
+        st.session_state.vectorstore = Chroma.from_documents(documents=splits,embedding=embeddings)
+        st.session_state['isran'] = True
+
+    if st.session_state['isran'] == True:
+        st.write("Embedding created")
+
+    def fdocs(docs):
+        return "\n\n".join(doc.page_content for doc in docs)
+
+    def llm(question,context):
+        formatted_prompt = f"Question: {question}\n\nContext:{context}"
+        response = ollama.chat(model='mistral', messages=[
+            {
+                'role': 'user',
+                'content': formatted_prompt
+            },
+            ])
+        return response['message']['content']
+
+
+
+    def rag_chain(question): 
+        retriever = st.session_state.vectorstore.as_retriever()
+        retrieved_docs = retriever.invoke(question)
+        formatted_context = fdocs(retrieved_docs)
+        return llm(question,formatted_context)
+
+    if 'messages' not in st.session_state:
+        st.session_state.messages = []
+
+    for message in st.session_state.messages:
+        st.chat_message(message['role']).markdown(message['content'])
+
+    prompt = st.chat_input("Say something")
+    response = rag_chain(prompt)
+    if prompt:
+        st.chat_message('user').markdown(prompt)
+        st.session_state.messages.append({'role':'user','content':prompt})
+        st.session_state.messages.append({'role':'AI','content':response})
+        st.chat_message('AI').markdown(response)

requirements.txt

@@ -0,0 +1,12 @@
+beautifulsoup4==4.12.3
+langchain==0.2.0
+langchain_community==0.2.0
+langchain_core==0.2.0
+numpy==1.26.4
+ollama==0.2.0
+opencv_python==4.9.0.80
+pandas==2.2.2
+streamlit==1.33.0
+streamlit_image_coordinates==0.1.6
+ultralytics==8.1.46
+chromadb