feat(back): Add new GroundingDINO model #47
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # @Copyright: CEA-LIST/DIASI/SIALV/LVA (2023) | |
| # @Author: CEA-LIST/DIASI/SIALV/LVA <[email protected]> | |
| # @License: CECILL-C | |
| # | |
| # This software is a collaborative computer program whose purpose is to | |
| # generate and explore labeled data for computer vision applications. | |
| # This software is governed by the CeCILL-C license under French law and | |
| # abiding by the rules of distribution of free software. You can use, | |
| # modify and/ or redistribute the software under the terms of the CeCILL-C | |
| # license as circulated by CEA, CNRS and INRIA at the following URL | |
| # | |
| # http://www.cecill.info | |
| name: Lint | |
| on: | |
| push: | |
| branches: | |
| - "develop" | |
| - "main" | |
| pull_request: | |
| branches: | |
| - "develop" | |
| permissions: | |
| contents: read | |
| jobs: | |
| # This job will lint backend code (Python) with Pylint | |
| lint_back: | |
| name: Backend | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| - name: Set up Python 3.10 | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.10" | |
| # Install PyTorch and TensorFlow CPU versions manually to prevent installing CUDA | |
| # Install GitHub models manually as they cannot be included in PyPI | |
| - name: Install dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| python -m pip install pylint | |
| python -m pip install torch~=2.2.0 torchaudio~=2.2.0 torchvision~=0.17.0 --index-url https://download.pytorch.org/whl/cpu | |
| python -m pip install tensorflow-cpu~=2.15.0 | |
| python -m pip install segment-anything@git+https://github.com/facebookresearch/segment-anything | |
| python -m pip install mobile-sam@git+https://github.com/ChaoningZhang/MobileSAM | |
| python -m pip install groundingdino@git+https://github.com/IDEA-Research/GroundingDINO | |
| python -m pip install . | |
| - name: Lint backend code with Pylint | |
| run: | | |
| pylint pixano_inference/ --rcfile .github/workflows/.pylintrc |