search

CVIR / CoMix

This repository contains the official implementation of CoMix (NeurIPS 2021) https://arxiv.org/pdf/2110.15128.pdf.
Apache License 2.0
20 stars 8 forks source link

How do you do Background Extraction for the Epic-Kitchens dataset? #1

Closed selva604 closed 2 years ago

selva604 commented 2 years ago

Hello. I'm trying to train CoMix for the trasfer task of D1-D2 from Epic-Kitchens dataset. Could you tell me example code for the Epic-Kitchens on extract_bg_tmf.py.

AadSah commented 2 years ago

Hi @selva604

Thanks for your interest. Background extraction for Epic-Kitchens requires some minor changes on reading the input file, as we now have .pkl instead of the .csv files. I have pushed a file with the required changes. You can find it here: https://github.com/CVIR/CoMix/blob/main/background_extraction/extract_bg_tmf_epickitchens.py

Thanks

selva604 commented 2 years ago

Thanks!