megalado
commited on
Commit
·
c13b92a
1
Parent(s):
c02fe08
Update app to use proper MDM checkpoint and setup
Browse files
app.py
CHANGED
|
@@ -214,7 +214,7 @@ def visualize_motion(output_path):
|
|
| 214 |
else:
|
| 215 |
action_type = "Moving"
|
| 216 |
|
| 217 |
-
ax.set_title(
|
| 218 |
return ax
|
| 219 |
|
| 220 |
# Create animation
|
|
@@ -225,7 +225,7 @@ def visualize_motion(output_path):
|
|
| 225 |
anim.save(output_path, writer='ffmpeg', fps=30)
|
| 226 |
plt.close()
|
| 227 |
|
| 228 |
-
print(
|
| 229 |
return output_path
|
| 230 |
|
| 231 |
# Create and visualize a motion
|
|
|
|
| 214 |
else:
|
| 215 |
action_type = "Moving"
|
| 216 |
|
| 217 |
+
ax.set_title(action_type + " Motion - Frame " + str(frame))
|
| 218 |
return ax
|
| 219 |
|
| 220 |
# Create animation
|
|
|
|
| 225 |
anim.save(output_path, writer='ffmpeg', fps=30)
|
| 226 |
plt.close()
|
| 227 |
|
| 228 |
+
print("Animation saved to " + output_path)
|
| 229 |
return output_path
|
| 230 |
|
| 231 |
# Create and visualize a motion
|