31 lines
771 B
Bash
31 lines
771 B
Bash
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
|
|
|
|
|