wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix videos in visualize_dataset are not in sync (#382)

This commit is contained in:
Mishig 2024-08-26 17:38:48 +02:00 committed by GitHub
parent 9ce98bb93c
commit aad59e6b6b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lerobot/templates/visualize_dataset_template.html
View File

@ -75,7 +75,7 @@
{% for video_info in videos_info %} {% for video_info in videos_info %}
<div class="max-w-96"> <div class="max-w-96">
<p class="text-sm text-gray-300 bg-gray-800 px-2 rounded-t-xl truncate">{{ video_info.filename }}</p> <p class="text-sm text-gray-300 bg-gray-800 px-2 rounded-t-xl truncate">{{ video_info.filename }}</p>
<video autoplay muted loop type="video/mp4" class="min-w-64" @timeupdate="() => { <video muted loop type="video/mp4" class="min-w-64" @canplay="videoCanPlay" @timeupdate="() => {
if (video.duration) { if (video.duration) {
const time = video.currentTime; const time = video.currentTime;
const pc = (100 / video.duration) * time; const pc = (100 / video.duration) * time;
@ -107,12 +107,12 @@
<!-- Controllers --> <!-- Controllers -->
<div class="flex gap-1 text-3xl items-center"> <div class="flex gap-1 text-3xl items-center">
<button x-ref="btnPlay" class="-rotate-90 hidden" class="-rotate-90" title="Play. Toggle with Space" @click="() => { <button x-ref="btnPlay" class="-rotate-90" class="-rotate-90" title="Play. Toggle with Space" @click="() => {
videos.forEach(video => video.play()); videos.forEach(video => video.play());
$refs.btnPlay.classList.toggle('hidden'); $refs.btnPlay.classList.toggle('hidden');
$refs.btnPause.classList.toggle('hidden'); $refs.btnPause.classList.toggle('hidden');
}">🔽</button> }">🔽</button>
<button x-ref="btnPause" title="Pause. Toggle with Space" @click="() => { <button x-ref="btnPause" class="hidden" title="Pause. Toggle with Space" @click="() => {
videos.forEach(video => video.pause()); videos.forEach(video => video.pause());
$refs.btnPlay.classList.toggle('hidden'); $refs.btnPlay.classList.toggle('hidden');
$refs.btnPause.classList.toggle('hidden'); $refs.btnPause.classList.toggle('hidden');
@ -125,7 +125,6 @@
@click="() => (videos.forEach(video => (video.currentTime = 0.0)))">↩️</button> @click="() => (videos.forEach(video => (video.currentTime = 0.0)))">↩️</button>
<input x-ref="slider" max="100" min="0" step="1" type="range" value="0" class="w-80 mx-2" @input="() => { <input x-ref="slider" max="100" min="0" step="1" type="range" value="0" class="w-80 mx-2" @input="() => {
const sliderValue = $refs.slider.value; const sliderValue = $refs.slider.value;
$refs.btnPause.click();
videos.forEach(video => { videos.forEach(video => {
const time = (video.duration * sliderValue) / 100; const time = (video.duration * sliderValue) / 100;
video.currentTime = time; video.currentTime = time;
@ -208,6 +207,8 @@
videos: null, videos: null,
video: null, video: null,
colors: null, colors: null,
nVideos: {{ videos_info | length }},
nVideoReadyToPlay: 0,
// alpine initialization // alpine initialization
init() { init() {
@ -343,7 +344,6 @@
window.history.replaceState({}, '', url.toString()); window.history.replaceState({}, '', url.toString());
}, },
formatTime(time) { formatTime(time) {
var hours = Math.floor(time / 3600); var hours = Math.floor(time / 3600);
var minutes = Math.floor((time % 3600) / 60); var minutes = Math.floor((time % 3600) / 60);
@ -351,6 +351,14 @@
return (hours > 0 ? hours + ':' : '') + (minutes < 10 ? '0' + minutes : minutes) + ':' + (seconds < return (hours > 0 ? hours + ':' : '') + (minutes < 10 ? '0' + minutes : minutes) + ':' + (seconds <
10 ? 10 ?
'0' + seconds : seconds); '0' + seconds : seconds);
},
videoCanPlay() {
this.nVideoReadyToPlay += 1;
if(this.nVideoReadyToPlay == this.nVideos) {
// start autoplay all videos in sync
this.$refs.btnPlay.click();
}
} }
}; };
} }