#!/bin/bash

# Test script for fish point cloud quality classification

# Set paths
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
POINTNET_DIR="${SCRIPT_DIR}/Pointnet_Pointnet2_pytorch"
TEST_FOLDER='/home/ubuntu/projects/FishMeasure/output_preview/HD1080_SN43186771_01-53-34/cloud'
LOG_DIR="fish_pointnet2_finetune"  # Change this to your trained model directory

cd "${POINTNET_DIR}"

# Test on folder of PLY files (no labels required)
python test_classification.py \
    --log_dir "${LOG_DIR}" \
    --test_folder "${TEST_FOLDER}" \
    --num_category 2 \
    --batch_size 24 \
    --num_votes 3 \
    --gpu 0 \
    --output batch_results.json

echo "Test completed!"