diff --git a/src/mouse_tracking/pytorch_inference/fecal_boli.py b/src/mouse_tracking/pytorch_inference/fecal_boli.py index 0b22ed1f..b169457d 100644 --- a/src/mouse_tracking/pytorch_inference/fecal_boli.py +++ b/src/mouse_tracking/pytorch_inference/fecal_boli.py @@ -163,12 +163,13 @@ def infer_fecal_boli_pytorch(args): ) final_fecal_boli_detections = fecal_boli_results.get_results() final_fecal_boli_counts = fecal_boli_counts.get_results() - write_fecal_boli_data( - args.out_file, - final_fecal_boli_detections, - final_fecal_boli_counts, - args.frame_interval, - model_definition["model-name"], - model_definition["model-checkpoint"], - ) + if args.out_file is not None: + write_fecal_boli_data( + args.out_file, + final_fecal_boli_detections, + final_fecal_boli_counts, + args.frame_interval, + model_definition["model-name"], + model_definition["model-checkpoint"], + ) performance_accumulator.print_performance() diff --git a/src/mouse_tracking/tfs_inference/arena_corners.py b/src/mouse_tracking/tfs_inference/arena_corners.py index 8314d2ad..cc9c98b3 100644 --- a/src/mouse_tracking/tfs_inference/arena_corners.py +++ b/src/mouse_tracking/tfs_inference/arena_corners.py @@ -113,7 +113,8 @@ def infer_arena_corner_model(args): model_definition["model-checkpoint"], ) px_per_cm = get_px_per_cm(filtered_corners) - write_pixel_per_cm_attr(args.out_file, px_per_cm, "corner_detection") + if args.out_file is not None: + write_pixel_per_cm_attr(args.out_file, px_per_cm, "corner_detection") if args.out_image is not None: render = plot_keypoints(filtered_corners, frame) imageio.imwrite(args.out_image, render)