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

Code for the head detector (HeadHunter) proposed in our CVPR 2021 paper Tracking Pedestrian Heads in Dense Crowd.

This repository provides the official code for GeNER (an automated dataset Generation framework for NER).

Training, generation, and analysis code for Learning Particle Physics by Example: Location-Aware Generative Adversarial Networks for Physics

Official PyTorch implementation of Data-free Knowledge Distillation for Object Detection, WACV 2021.

A repository for the updated version of CoinRun used to collect MUGEN, a multimodal video-audio-text dataset.

This repository contains a pytorch implementation of "StereoPIFu: Depth Aware Clothed Human Digitization via Stereo Vision".

League of Legends Reinforcement Learning Environment (LoLRLE) multiple training scenarios using PPO.

Final project code: Implementing BicycleGAN, for CIS680 FA21 at University of Pennsylvania

Deep Learning Specialization by Andrew Ng, deeplearning.ai.

ICNet for Real-Time Semantic Segmentation on High-Resolution Images, ECCV2018

Image to Image translation, image generataton, few shot learning

Official Code For TDEER: An Efficient Translating Decoding Schema for Joint Extraction of Entities and Relations (EMNLP2021)

Official repository for Hierarchical Opacity Propagation for Image Matting

Spatio-Temporal Entropy Model (STEM) for end-to-end leaned video compression.

PaddleViT: State-of-the-art Visual Transformer and MLP Models for PaddlePaddle 2.0+

Single-stage Keypoint-based Category-level Object Pose Estimation from an RGB Image

This toolkit provides codes to download and pre-process the SLUE datasets, train the baseline models, and evaluate SLUE tasks.

[ICML 2021] DouZero: Mastering DouDizhu with Self-Play Deep Reinforcement Learning | 斗地主AI

A Convolutional Transformer for Keyword Spotting

Machine learning framework for both deep learning and traditional algorithms