Extracting Masks

[IsaiasBahena]

Can a feature be added that allows you to extract the masks from the videos as well?

[KDeng98]

Hi there,
If you are still looking for it, here is a tip of enabling it in a previous issue.
#26

Here, you can uncomment this line to save the maks.

Track-Anything/app.py

Line 380 in 57c28bc

# args.mask_save = True

And you can change the mask save directory here.

Track-Anything/app.py

Lines 278 to 285 in 57c28bc

	if interactive_state["mask_save"]:
	if not os.path.exists('./result/mask/{}'.format(video_state["video_name"].split('.')[0])):
	os.makedirs('./result/mask/{}'.format(video_state["video_name"].split('.')[0]))
	i = 0
	print("save mask")
	for mask in video_state["masks"]:
	np.save(os.path.join('./result/mask/{}'.format(video_state["video_name"].split('.')[0]), '{:05d}.npy'.format(i)), mask)
	i+=1