Closed peterklipfel closed 1 year ago
Befor you execute the precompute_knn.py, You should execute the crop_datasets.py and then it will auto create the floder. Path is cropped\cocostuff27_five_crop_0.5.
Thank you very much @KennyChen880127! I've submitted a pull request to document this in the readme. Feel free to close after accepting/rejecting the PR
Thank you for submitting the PR! I have just merged it in (Also sorry for the delay in getting back to all the issues!). Closing this out for now
I am trying to run the training of the model manually (to identify OOM issues for custom training), but when I follow the steps in the readme, I get an error when looking for images in a folder that doesn't exist. There are several possible issues here, so I'll try to detail them independently.
Steps to reproduce:
src/configs/train_config.yml
andsrc/configs/eval_config.yml
with my pytorch data directorypython download_datasets.py
cd $PYTORCH_DATA_DIR
cd $STEGO_DIR/src
python precompute_knns.py
I get no results when I run:
Other things that I don't understand that may be related:
/home/peter/datasets/cropped/cocostuff27_five_crop_0.5/img/val
, but I would expect/home/peter/datasets/SOME_DATASET_NAME/cropped/cocostuff27_five_crop_0.5/img/val
27
, but there are several references to2017
. Perhaps there is a typo here and the code just needs to be updated to 2017 (i.e.train_2017
andval_2017
?