mirror of
https://github.com/gaomingqi/Track-Anything.git
synced 2025-12-16 08:27:49 +01:00
17 lines
457 B
Python
17 lines
457 B
Python
|
"""
|
||
|
|
load_subset.py - Presents a subset of data
|
||
|
|
DAVIS - only the training set
|
||
|
|
YouTubeVOS - I manually filtered some erroneous ones out but I haven't checked all
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
def load_sub_davis(path='util/davis_subset.txt'):
|
||
|
|
with open(path, mode='r') as f:
|
||
|
|
subset = set(f.read().splitlines())
|
||
|
|
return subset
|
||
|
|
|
||
|
|
def load_sub_yv(path='util/yv_subset.txt'):
|
||
|
|
with open(path, mode='r') as f:
|
||
|
|
subset = set(f.read().splitlines())
|
||
|
|
return subset
|