Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,216 Bytes
6073e55 23fdbc0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
# Copyright (c) 2025 Ye Liu. Licensed under the BSD-3-Clause License.
import nncore
from videomind.dataset.hybrid import DATASETS
from videomind.dataset.wrappers import GroundingDataset
from videomind.utils.parser import parse_query
@DATASETS.register(name='vid_morp')
class VidMorpDataset(GroundingDataset):
ANNO_PATH = 'data/vid_morp/anno_vid_morp.jsonl'
VIDEO_ROOT = 'data/vid_morp/videos_3fps_480_noaudio'
UNIT = 0.001
@classmethod
def load_annos(self, split='train'):
assert split == 'train'
raw_annos = nncore.load(self.ANNO_PATH)
all_videos = nncore.ls(self.VIDEO_ROOT, ext='.mp4')
all_videos = set(v[:11] for v in all_videos)
annos = []
for raw_anno in raw_annos:
vid = raw_anno['vid']
if vid not in all_videos:
continue
anno = dict(
source='vid_morp',
data_type='grounding',
video_path=nncore.join(self.VIDEO_ROOT, vid + '.mp4'),
duration=raw_anno['duration'],
query=parse_query(raw_anno['query']),
span=[raw_anno['span']])
annos.append(anno)
return annos
|