yolov8_cspdarknet-detection

训练模型

python trainer.py --device_id cuda \
--pretrained_pt_path /res/pvdetection_07fe2b.pt \
--batch_size 16 \
--max_epochs 300 \
--ema_enabled \
--wandb_enabled \
--early_stop \
--trainset_path /mypath/quaming_veh_dataset/images/train \
--valset_path /mypath/quaming_veh_dataset/images/val

运行demo

python demo.py --weight my_weight.pt --camera

导出模型

python export.py --weight my_weight.pt --input_shape 1 3 1280 1280 --input_names image --output_names box0 cls0 box1 cls1 box2 cls2 --opset_version 13 --enable_onnxsim

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
res		res
.DS_Store		.DS_Store
box.py		box.py
canvas.py		canvas.py
dataloader.py		dataloader.py
dataset.py		dataset.py
dataset_visualizer.py		dataset_visualizer.py
demo.py		demo.py
ema.py		ema.py
export.py		export.py
general.py		general.py
model.py		model.py
model_loss.py		model_loss.py
nms.py		nms.py
readme.md		readme.md
trainer.py		trainer.py
transforms.py		transforms.py

klrc/yolov8_cspdarknet-object_detection

Folders and files

Latest commit

History

Repository files navigation

yolov8_cspdarknet-detection

训练模型

运行demo

导出模型

About

Resources

Stars

Watchers

Forks

Languages