- # For 用于训练不同模型，以及保存相应的路径
- # -------------------------------------------------------------------------------------------------------------------- #
- python run.py --model 'yolov7' --save_path './checkpoints/yolov7/watermarking/best.pt' --save_path_last './checkpoints/yolov7/watermarking/last.pt' --epoch 100
- # For 用于剪枝模型，剪枝后微调训练，保存剪枝后模型路径，以及验证微调模型准确性
- # -------------------------------------------------------------------------------------------------------------------- #
- python run.py --model 'yolov7' --prune True --prune_weight './checkpoints/yolov7/watermarking/best.pt' --prune_save './checkpoints/yolov7/watermarking/prune_best.pt' --epoch 40
|
