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

Emotion Recognition from Facial Images

Acute ischemic stroke dataset

Embeddinghub is a database built for machine learning embeddings.

Keras-retinanet - Keras implementation of RetinaNet object detection.

Differential fuzzing for the masses!

Unofficial Tensorflow Implementation of ConvNeXt from A ConvNet for the 2020s

Making a music video with Wav2CLIP and VQGAN-CLIP

Code for "Adversarial Attack Generation Empowered by Min-Max Optimization", NeurIPS 2021

BADet: Boundary-Aware 3D Object Detection from Point Clouds (Pattern Recognition 2022)

Code for KHGT model, AAAI2021

PyTorch implementation of SwAV (Swapping Assignments between Views)

Code for the paper: Learning Adversarially Robust Representations via Worst-Case Mutual Information Maximization (https://arxiv.org/abs/2002.11798)

Code to accompany our paper "Continual Learning Through Synaptic Intelligence" ICML 2017

A collection of educational notebooks on multi-view geometry and computer vision.

A very simple baseline to estimate 2D & 3D SMPL-compatible keypoints from a single color image.

T2F: text to face generation using Deep Learning

PyTorch GPU implementation of the ES-RNN model for time series forecasting

I tried to apply the CAM algorithm to YOLOv4 and it worked.

[ICCV 2021] Official PyTorch implementation for Deep Relational Metric Learning.

AdamW optimizer and cosine learning rate annealing with restarts