hkchengrex / MiVOS

Hi,

In the fusion_dataset.py , it may has a bug.

# We need the second reference frame to be visible from the first
if not path.exists(path.join(video_path, first_ref, tar_obj, r+'.png')):
    continue
# We need the target object to exist
if path.exists(path.join(video_path, r, tar_obj, tar_frame)):
    src2_ref_options.append(r)

Should it be like this here? Because tar_frame is a number like "00001".

if path.exists(path.join(video_path, r, tar_obj, tar_frame+'.png')):

I must admit that the naming is not exactly clear... Sorry about that

tar_frame is the file name while tar_frame_int is the integer.

MiVOS/dataset/fusion_dataset.py

Line 107 in 210e76f

    
           tar_frame = np.random.choice(os.listdir(path.join(video_path, first_ref, tar_obj)))

MiVOS/dataset/fusion_dataset.py

Line 109 in 210e76f

tar_frame_int = int(tar_frame[:-4])

Thank you. It is my problem...

A possible bug？