Spaces:
Sleeping
Sleeping
Yichuan Huang
commited on
Upload 9 files
Browse files- .gitattributes +3 -0
- app.py +19 -0
- models/best_model.zip +3 -0
- models/final_model.zip +3 -0
- rendered_videos/car_racing_best_model.avi +3 -0
- rendered_videos/car_racing_best_model.mp4 +0 -0
- rendered_videos/car_racing_final_model.avi +3 -0
- rendered_videos/car_racing_final_model.mp4 +3 -0
- requirements.txt +9 -0
- viewer.py +9 -0
.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
|