This is a PyTorch re-implementation to show the preprocessing of feature computation and clustering can be done on the fly. (in progress)
Run a demo training code for coseg.
-
Download the COSEG. A script can be used to download the dataset.
-
Put the dataset in
../dataset
and make a working directory in../work_dir
-
Runing the training code
python train_coseg