Bling's Object detection tool

Last update: Nov 01, 2022

Related tags

Deep Learning BriVL-BUA-applications

Overview

BriVL for Building Applications

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 this Pipeline

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

Download Models

bua-caffe-frcn-r101_with_attributes.pth -> /bbox_extractor/weights
chinese-roberta-wwm-ext -> /BriVL/weights/hfl
tf_efficientnet_b5_ns-6f26d0cf.pth -> /BriVL/weights
brivl-weights.pth* -> /BriVL/weights

Requirements

Python >= 3.6
PyTorch >= 1.4
Cuda >= 9.2 and cuDNN
Detectron2 <= 0.3
Transformers

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 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu110/torch1.7/index.html`
10.2	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.7/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.6/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu102/torch1.5/index.html`
10.1	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.7/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.6/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu101/torch1.5/index.html`
9.2	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.7/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.6/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cu92/torch1.5/index.html`
cpu	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.7/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.6/index.html`	install `python -m pip install detectron2 -f \ https://dl.fbaipublicfiles.com/detectron2/wheels/cpu/torch1.5/index.html`

More Resources

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.

Contact

This repo is maintained by Chuhao JIn(@jinchuhao).

Bling's Object detection tool

Related tags

Overview

BriVL for Building Applications

This repo is re-implemented from following projects:

Online Demo built by BriVL

Contents

Test this Pipeline

Download Models

Requirements

More Resources

Contact

Owner

chuhaojin

Official Implementation of Swapping Autoencoder for Deep Image Manipulation (NeurIPS 2020)

Invariant Causal Prediction for Block MDPs

Official PyTorch implementation of BlobGAN: Spatially Disentangled Scene Representations

Python library containing BART query generation and BERT-based Siamese models for neural retrieval.

Rethinking Transformer-based Set Prediction for Object Detection

Torchyolo - Yolov3 ve Yolov4 modellerin Pytorch uygulamasıdır

EgoNN: Egocentric Neural Network for Point Cloud Based 6DoF Relocalization at the City Scale

Symmetry and Uncertainty-Aware Object SLAM for 6DoF Object Pose Estimation

Pytorch implementation of RED-SDS (NeurIPS 2021).

DvD-TD3: Diversity via Determinants for TD3 version

Reproducing-BowNet: Learning Representations by Predicting Bags of Visual Words

This repo is a PyTorch implementation for Paper "Unsupervised Learning for Cuboid Shape Abstraction via Joint Segmentation from Point Clouds"

The Rich Get Richer: Disparate Impact of Semi-Supervised Learning

The datasets and code of ACL 2021 paper "Aspect-Category-Opinion-Sentiment Quadruple Extraction with Implicit Aspects and Opinions".

(ICCV 2021) ProHMR - Probabilistic Modeling for Human Mesh Recovery

OpenVINO黑客松比赛项目

Fluency ENhanced Sentence-bert Evaluation (FENSE), metric for audio caption evaluation. And Benchmark dataset AudioCaps-Eval, Clotho-Eval.

QuadTree Attention for Vision Transformers (ICLR2022)

object recognition with machine learning on Respberry pi

Code for "The Box Size Confidence Bias Harms Your Object Detector"