Files
FishServer/FishMeasure/run_fish_evaluation_batch.sh
2026-05-06 15:59:38 +08:00

20 lines
915 B
Bash

#!/bin/bash
# Simple command to run fish video weight evaluation with all flags
python3 fish_video_weight_evaluation.py \
--yolo-model "/home/ubuntu/projects/FishMeasure/runs/train/fish_detection_20251127_104658/weights/best.pt" \
--batch-svo-folder "/home/ubuntu/data/fish/2016-1-22-last/fish19" \
--save-output "output_test" \
--conf 0.5 \
--imgsz 640 \
--sam-device "cuda" \
--max-frames 0 \
--filter-pointcloud \
--use-density-filter \
--pointcloud-classifier "/home/ubuntu/projects/FishMeasure/pointcloud_classifier/Pointnet_Pointnet2_pytorch/log/classification/fish_pointnet2_finetune/checkpoints/best_model.pth" \
--use-pointcloud-classifier \
--pointcloud-classifier-threshold 0.7 \
--run-weight-estimation \
--weight-estimator-checkpoint /home/ubuntu/projects/FishMeasure/weight_estimator/runs/dgcnn_20260312_171043/best.pt \
--weight-top-k 5