LibrePICODETs

PicoDet-S (ESNet backbone, CSP-PAN neck, GFL/DFL head), repackaged for LibreYOLO.

Source

Derived from Bo396543018/Picodet_Pytorch, which itself ports PaddlePaddle's PicoDet to PyTorch via mmdet/mmcv. Copyright (c) 2018-2023 OpenMMLab. Licensed under the Apache License, Version 2.0.

Modifications

State-dict key remapping only: bbox_head.* -> head.*, backbone.<stage>_<i>.* -> backbone.blocks.<flat>.*, neck.trans.trans.* -> neck.trans.*, and unwrapping mmcv's ConvModule inside SE layers. Learned parameters are unchanged. See weights/convert_picodet_weights.py in the LibreYOLO source repository.

Usage

pip install libreyolo

from libreyolo import LibreYOLO

model = LibreYOLO("LibrePICODETs.pt")
results = model("image.jpg", save=True)

License

Apache License 2.0. See the LICENSE and NOTICE files in this repository.