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

Commit

Permalink
include recall
Browse files Browse the repository at this point in the history
  • Loading branch information
bw4sz committed Apr 5, 2019
1 parent 216d8c7 commit 1cd7c23
Showing 1 changed file with 10 additions and 10 deletions.
20 changes: 10 additions & 10 deletions eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -131,18 +131,18 @@ def main(data, DeepForest_config, experiment, args=None):
NEON_generator = create_NEON_generator(args.batch_size, DeepForest_config)
NEON_recall_generator = create_NEON_generator(args.batch_size, DeepForest_config)

#recall = neonRecall(
#sites,
#NEON_recall_generator,
#model,
#score_threshold=args.score_threshold,
#save_path=args.save_path + dirname,
#max_detections=args.max_detections
#)
recall = neonRecall(
sites,
NEON_recall_generator,
model,
score_threshold=args.score_threshold,
save_path=args.save_path + dirname,
max_detections=args.max_detections
)

#print("Recall is {:0.3f}".format(recall))
print("Recall is {:0.3f}".format(recall))

#experiment.log_metric("Recall", recall)
experiment.log_metric("Recall", recall)

#NEON plot mAP
average_precisions = evaluate(
Expand Down

0 comments on commit 1cd7c23

Please sign in to comment.