BriVL for application development
This repo is used for illustrating how to build applications by using BriVL model.
This repo is re-implemented from following projects:
Online Demo built by BriVL
This repo contains two parts:
- Bounding Box Extractor:
./bbox_extractor
- BriVL Feature Extractor:
./BriVL
Test image has been saved in ./bbox_extractor/feature_extractor
, test with following command:
python3 main.py --brivl_cfg BriVL/cfg/BriVL_cfg.yml --brivl_weights BriVL/weights/brivl-weights.pth
Important: The version of Detectron2 should be 0.3 or below.
Install Pre-Built Detectron2 (Linux only)
Choose from this table to install v0.3 (Nov 2020):
CUDA | torch 1.7 | torch 1.6 | torch 1.5 |
---|
11.0 | install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu110/torch1.7/index.html
| | |
10.2 | install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.7/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.6/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.5/index.html
|
10.1 | install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.7/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.6/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.5/index.html
|
9.2 | install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.7/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.6/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.5/index.html
|
cpu | install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.7/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.6/index.html
| install python -m pip install detectron2==0.3 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.5/index.html
|
Source Code of BriVL 1.0
Model of BriVL 1.0*
Online API of BriVL 1.0
Online API of BriVL 2.0
* indicates an application is needed.
This repo is maintained by Chuhao JIn(@jinchuhao).