54 lines
1.7 KiB
Bash
54 lines
1.7 KiB
Bash
|
#!/usr/bin/env bash
|
||
|
|
# Single case: fish9 only (same flags as run_predict_from_svo2.sh).
|
||
|
|
# Layout: output_weight_estimator/fish9/<svo_stem>/{cloud,images,weight_prediction.json,...}
|
||
|
|
#
|
||
|
|
if [ -z "${BASH_VERSION:-}" ]; then
|
||
|
|
exec bash "$0" "$@"
|
||
|
|
fi
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
cd "$SCRIPT_DIR"
|
||
|
|
|
||
|
|
SESSION_ROOT="/home/ubuntu/data/fish/2016-1-22-last"
|
||
|
|
FISH_NAME="fish17"
|
||
|
|
fish_dir="${SESSION_ROOT}/${FISH_NAME}/"
|
||
|
|
OUT_PARENT="output_weight_estimator"
|
||
|
|
save_out="${OUT_PARENT}/${FISH_NAME}"
|
||
|
|
|
||
|
|
if [[ ! -d "$fish_dir" ]]; then
|
||
|
|
echo "ERROR: not a directory: $fish_dir" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
shopt -s nullglob
|
||
|
|
svos=("${fish_dir}"*.svo2)
|
||
|
|
shopt -u nullglob
|
||
|
|
if [[ ${#svos[@]} -eq 0 ]]; then
|
||
|
|
echo "ERROR: no .svo2 in $fish_dir" >&2
|
||
|
|
exit 1
|
||
|
|
fi
|
||
|
|
|
||
|
|
echo "============================================================"
|
||
|
|
echo " ${FISH_NAME} (${#svos[@]} .svo2) → ${save_out}/<svo_stem>/"
|
||
|
|
echo "============================================================"
|
||
|
|
|
||
|
|
python3 predict_weigth_from_svo2.py \
|
||
|
|
--batch-svo-folder "$fish_dir" \
|
||
|
|
--weight-checkpoint weight_estimator/runs/dgcnn_20260312_171043/best.pt \
|
||
|
|
--save-output "$save_out" \
|
||
|
|
--yolo-model "/home/ubuntu/projects/FishMeasure/runs/train/fish_detection_20251127_104658/weights/best.pt" \
|
||
|
|
--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 \
|
||
|
|
--use-flatness-filter \
|
||
|
|
--flatness-threshold 55.0 \
|
||
|
|
--frame-stride 1 \
|
||
|
|
--weight-top-k 5 \
|
||
|
|
--weight-top-by-length
|