Yichuan Huang commited on
Commit
2d52e99
·
verified ·
1 Parent(s): c912fed

Upload 9 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ rendered_videos/car_racing_best_model.avi filter=lfs diff=lfs merge=lfs -text
37
+ rendered_videos/car_racing_final_model.avi filter=lfs diff=lfs merge=lfs -text
38
+ rendered_videos/car_racing_final_model.mp4 filter=lfs diff=lfs merge=lfs -text
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from viewer import show_videos_and_scores
3
+
4
+ # Define the Gradio interface
5
+ interface = gr.Interface(
6
+ fn=show_videos_and_scores,
7
+ inputs=[],
8
+ outputs=[
9
+ gr.Video(label="Final Model Video", value="./rendered_videos/car_racing_final_model.avi"),
10
+ gr.Textbox(label="Final Model Score", value="902.5"),
11
+ gr.Video(label="Best Model Video", value="./rendered_videos/car_racing_best_model.avi"),
12
+ gr.Textbox(label="Best Model Score", value="919.9"),
13
+ ],
14
+ title="Car Racing Videos Demo",
15
+ description="This demo shows the final and best model videos and scores of the Car Racing environment.",
16
+ )
17
+
18
+ if __name__ == "__main__":
19
+ interface.launch()
models/best_model.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c5746c16fc44a2557fc842f19958abda2859a2ab390e4b77785b5fce55669384
3
+ size 35075182
models/final_model.zip ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:40ea1feece372349ba312e114686ccd9df647d6641e64be97de5fbdb1f590c90
3
+ size 35075165
rendered_videos/car_racing_best_model.avi ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:508aadba94d69da3a29404c46bc5d08c5ffe95f6c2c858bdc9f10c9070a72b70
3
+ size 10267326
rendered_videos/car_racing_best_model.mp4 ADDED
Binary file (961 kB). View file
 
rendered_videos/car_racing_final_model.avi ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b5ff2f070b53052ff4befa5131f7b8faa257c5c2a2d7f1b4994b43dad5951730
3
+ size 10803816
rendered_videos/car_racing_final_model.mp4 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:48584f566a6f16fb1df904a81983131b956daaa7014463a15df6451d3416d64b
3
+ size 1049375
requirements.txt ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ swig
2
+ gymnasium
3
+ gymnasium[box2d]
4
+ stable-baselines3
5
+ torch
6
+ torchvision
7
+ numpy
8
+ opencv-python==4.5.5.64
9
+ tensorboard
viewer.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ def show_videos_and_scores():
2
+ # Paths to the video files
3
+ final_model_video = "./rendered_videos/car_racing_final_model.avi"
4
+ best_model_video = "./rendered_videos/car_racing_best_model.avi"
5
+ # Scores of the videos, this can be found in tensorboard logs
6
+ final_model_score = 902.5
7
+ best_model_score = 919.9
8
+
9
+ return final_model_video, final_model_score, best_model_video, best_model_score