代码拉取完成,页面将自动刷新
同步操作将从 DeepSpark/DeepSparkHub 强制同步,此操作会覆盖自 Fork 仓库以来所做的任何修改,且无法恢复!!!
确定后同步将在后台操作,完成时将刷新页面,请耐心等待。
PicoDet is an ultra lightweight real-time object detection model that includes four different sizes of XS/S/M/L.
By using structures such as TAL, ETA Head, and PAN, the accuracy of the model is improved. When deploying model inference,
the model supports including post-processing in the network, thereby supporting direct output of prediction results.
git clone https://github.com/PaddlePaddle/PaddleDetection.git
cd PaddleDetection
pip3 install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
python3 setup.py install
pip3 install protobuf==3.20.3
pip3 install numba==0.56.4
yum install mesa-libGl -y
Download the Coco datasets, Specify coco2017 to your Coco path in later training process.
The Coco datesets path structure should look like(assuming coco2017 is used):
coco2017
├── annotations
│ ├── instances_train2017.json
│ ├── instances_val2017.json
│ └── ...
├── train2017
│ ├── 000000000009.jpg
│ ├── 000000000025.jpg
│ └── ...
├── val2017
│ ├── 000000000139.jpg
│ ├── 000000000285.jpg
│ └── ...
├── train2017.txt
├── val2017.txt
└── ...
assuming we are going to train picodet-l, the model config file is 'configs/picodet/picodet_l_640_coco_lcnet.yml' vim configs/datasets/coco_detection.yml, set 'dataset_dir' in the configuration file to coco2017, then start trainging.
single gpu:
export CUDA_VISIBLE_DEVICES=0
python3 tools/train.py -c configs/picodet/picodet_l_640_coco_lcnet.yml --eval
multi-gpu:
export CUDA_VISIBLE_DEVICES=0,1,2,3,4,5,6,7
python3 -m paddle.distributed.launch --gpus 0,1,2,3,4,5,6,7 tools/train.py -c configs/picodet/picodet_l_640_coco_lcnet.yml --eval
GPUs | IPS | mAP0.5:0.95 | mAP0.5 |
---|---|---|---|
BI-V100 x 8 | 19.84 | 41.2 | 58.2 |
https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.6/configs/picodet/README_en.md
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。