Skip to content
This repository has been archived by the owner on Jan 9, 2024. It is now read-only.

Commit

Permalink
Save path
Browse files Browse the repository at this point in the history
  • Loading branch information
bw4sz committed Jul 22, 2019
1 parent 000855b commit bd0d9a9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions analysis/predict_OSBS.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,8 @@ def main(DeepForest_config, model=None, args=None):
#save time for logging
dirname = datetime.now().strftime("%Y%m%d_%H%M%S")
save_path = DeepForest_config["save_image_path"] + dirname

print("save path is {}".format(save_path))

#Evaluation metrics
site = DeepForest_config["evaluation_site"]

Expand Down Expand Up @@ -93,7 +94,6 @@ def main(DeepForest_config, model=None, args=None):
"--batch-size", str(DeepForest_config['batch_size']),
'--score-threshold', str(score_threshold),
'--suppression-threshold','0.15',
'--save-path', save_path,
'--model', trained_model,
'--convert-model'
]
Expand Down

0 comments on commit bd0d9a9

Please sign in to comment.