Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
Score threshold
  • Loading branch information
Ben Weinstein authored and Ben Weinstein committed Feb 27, 2021
1 parent 35d0fbf commit b2bf25a
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions deepforest_config.yml
Expand Up @@ -7,7 +7,7 @@ gpus: 3
distributed_backend: ddp
batch_size: 5
nms_thresh: 0.05

score_thresh: 0.2
train:
csv_file: /orange/ewhite/b.weinstein/NeonTreeEvaluation/hand_annotations/crops/hand_annotations.csv
root_dir: /orange/ewhite/b.weinstein/NeonTreeEvaluation/hand_annotations/crops/
Expand All @@ -28,5 +28,4 @@ validation:
root_dir: /home/b.weinstein/NeonTreeEvaluation/evaluation/RGB/
project: False
#Intersection over union evaluation
iou_threshold: 0.4
score_threshold: 0.05
iou_threshold: 0.4

0 comments on commit b2bf25a

Please sign in to comment.