search

SaoYan / bgsCNN

Background subtraction using deep learning method.
GNU General Public License v3.0
191 stars 61 forks source link

log_dir needed? #10

Closed blackmad closed 3 years ago

blackmad commented 6 years ago

Hey,

I'm trying to understand if log_dir is needed when using a pre-trained model. All of the scripts (including result.py, which seems like just the "run the model" script) try to load a checkpoint from there, and fail if it's not there. Is that required? Should that be included with the pre-trained model? Can I just delete that session restore?

Or do I really need to run my own training on the dataset with the model?

Thanks, David

SaoYan commented 5 years ago

Hi,

log_dir is used for:

  1. during training, store all the log info, including checkpoints, tensorboard info, etc.
  2. during test, the checkpoints (pre-trained model parameters) are restored from this dir; also, a separate tensorboard visualization is stored under sub-dir "model_test", this include image/seg visualization for the convenience of debugging

I realize this is a huge delay... hope this still helps