cd Pointnet_Pointnet2_pytorch

# # 使用 PointNet 训练
# python train_classification.py \
#     --use_fish_dataset \
#     --data_path ../../dataset/ \
#     --model pointnet_cls \
#     --num_category 2 \
#     --num_point 1024 \
#     --batch_size 24 \
#     --epoch 200 \
#     --learning_rate 0.001 \
#     --gpu 0 \
#     --log_dir fish_pointnet

# 使用 PointNet++ 训练
python train_classification.py \
    --use_fish_dataset \
    --data_path ../pc_quality_dataset/ \
    --model pointnet2_cls_ssg \
    --num_category 2 \
    --num_point 1024 \
    --batch_size 24 \
    --epoch 200 \
    --learning_rate 0.001 \
    --gpu 0 \
    --pretrained log/classification/pointnet2_ssg_wo_normals/checkpoints/best_model.pth \
    --log_dir fish_pointnet2_finetune