- 模型路径,名字为模型的文件夹,batchsize等参数都在这里修改:E:\paddle\PaddleDetection\configs
- 运行时是否使用GPU、保存路径等参数的修改(全部模型):E:\paddle\PaddleDetection\configs\runtime.yml
- 设置数据集路径,根据数据集类型来修改对应的yml:E:\paddle\PaddleDetection\configs\datasets
- 数据集存放路径,根据数据集类型放好:E:\paddle\PaddleDetection\dataset
- 训练:
文件:E:\paddle\PaddleDetection\tools\train.py
命令:
python train.py -c ..\configs\picodet\picodet_s_416_coco_lcnet.yml --eval
- 评估:
文件:E:\paddle\PaddleDetection\tools\eval.py
命令:
python eval.py -c ..\configs\picodet\picodet_s_416_coco_lcnet.yml -o weights=output\best_model\model.pdparams
- 预测:
文件:E:\paddle\PaddleDetection\tools\infer.py
命令:
python infer.py -c ..\configs\picodet\picodet_s_416_coco_lcnet.yml --infer_dir=../dataset/coco/dataset_cell/images --output_dir=infer_output/ --draw_threshold=0.5 -o weights=output\best_model\model.pdparams --use_vdl=True