24 lines
683 B
Bash
24 lines
683 B
Bash
#!/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!"
|