Code for "Sparse Steerable Convolutions: An Efficient Learning of SE(3)-Equivariant Features for Estimation and Tracking of Object Poses in 3D Space"

Overview

Sparse Steerable Convolution (SS-Conv)

Code for "Sparse Steerable Convolutions: An Efficient Learning of SE(3)-Equivariant Features for Estimation and Tracking of Object Poses in 3D Space", NeurIPS 2021.

Created by Jiehong Lin, Hongyang Li, Ke Chen, Jiangbo Lu, and Kui Jia.

You might also like...
Opinionated code formatter, just like Python's black code formatter but for Beancount

beancount-black Opinionated code formatter, just like Python's black code formatter but for Beancount Try it out online here Features MIT licensed - b

a delightful machine learning tool that allows you to train, test and use models without writing code
a delightful machine learning tool that allows you to train, test and use models without writing code

igel A delightful machine learning tool that allows you to train/fit, test and use models without writing code Note I'm also working on a GUI desktop

Pytorch Lightning code guideline for conferences

Deep learning project seed Use this seed to start new deep learning / ML projects. Built in setup.py Built in requirements Examples with MNIST Badges

Automatically Build Multiple ML Models with a Single Line of Code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.
Automatically Build Multiple ML Models with a Single Line of Code. Created by Ram Seshadri. Collaborators Welcome. Permission Granted upon Request.

Auto-ViML Automatically Build Variant Interpretable ML models fast! Auto_ViML is pronounced "auto vimal" (autovimal logo created by Sanket Ghanmare) N

Code samples for my book "Neural Networks and Deep Learning"

Code samples for "Neural Networks and Deep Learning" This repository contains code samples for my book on "Neural Networks and Deep Learning". The cod

Code for: https://berkeleyautomation.github.io/bags/

DeformableRavens Code for the paper Learning to Rearrange Deformable Cables, Fabrics, and Bags with Goal-Conditioned Transporter Networks. Here is the

Code for our method RePRI for Few-Shot Segmentation. Paper at http://arxiv.org/abs/2012.06166
Code for our method RePRI for Few-Shot Segmentation. Paper at http://arxiv.org/abs/2012.06166

Region Proportion Regularized Inference (RePRI) for Few-Shot Segmentation In this repo, we provide the code for our paper : "Few-Shot Segmentation Wit

Applications using the GTN library and code to reproduce experiments in "Differentiable Weighted Finite-State Transducers"

gtn_applications An applications library using GTN. Current examples include: Offline handwriting recognition Automatic speech recognition Installing

Code for
Code for "Contextual Non-Local Alignment over Full-Scale Representation for Text-Based Person Search"

Contextual Non-Local Alignment over Full-Scale Representation for Text-Based Person Search This is an implementation for our paper Contextual Non-Loca

Comments
  • isnt it correct to set optimizer and lr_scheduler's

    isnt it correct to set optimizer and lr_scheduler's "name" to "type" ?

    Hi~! I'm trying to start training but I got an simple error " KeyError: "cfg or default_args must contain the key 'type', but got {'name': 'Adam', 'lr': 0.01, 'betas': [0.5, 0.999], 'eps': 1e-06, 'params': [{'params': <filter object at 0x7fca95bd3f98>}]}\nNone"

    So I changed "name" to "type" of "optimizer" and then I got another error. KeyError: "cfg or default_args must contain the key 'type', but got {'name': 'StepLR', 'step_size': 2, 'gamma': 0.5, 'optimizer': Adam (\nParameter Group 0\n amsgrad: False\n betas: [0.5, 0.999]\n eps: 1e-06\n lr: 0.01\n weight_decay: 0\n)}\nNone"

    So, I again changed "name" to "type" of "lr_scheduler".

    and then it seems no more error but, there is no progress on trainning. just got stuck from scratch.

    2022-05-25 11:38:49,545 - using gpu: 2 2022-05-25 11:38:49,546 - => creating model ... 580 images found.

    I used my custom images/depth/ so on which are fit to NOCS format and I already passed running the data_processing.py script.

    Please help me~!

    opened by dedoogong 2
  • [install] header file missing

    [install] header file missing

    Hi, Appreciate your work,

    when I install SS_Conv_lib, it seems some header files are not found, SS-Conv/SS_Conv_lib/src/conv/geometry.h:16:10: fatal error: tensorview.h: No such file or directory SS-Conv/SS_Conv_lib/src/conv/reordering_functor.h:12:10: fatal error: tensorview.h: No such file or directory SS_Conv_lib/src/pool/global_maxpool.h:16:10: fatal error: datatype.h: No such file or directory

    any extra dependencies?

    opened by David-Willo 1
  • When will the code be released?

    When will the code be released?

    Hi, authors

    Your work is novel and I believe it will have a big impact on object pose estimation society. But I'm wondering if you are going to release the code in near future.

    opened by shanice-l 1
  • segmentation mask of LineMOD

    segmentation mask of LineMOD

    Very good work! I have a question about the LineMOD segmentation masks. You mentioned that the predicted masks provided by the PoseCNN[10] paper are used for testing in the supplementary material, but, as I know, PoseCNN paper does not provide segmentation masks for LineMOD neither on their project page nor their Github page. Is it possible to clarify how or where to obtain the predicted object masks, if I am wrong? Thanks!

    opened by dingdingcai 0
Owner
Research lab focusing on CV, ML, and AI
null
Inference code for "StylePeople: A Generative Model of Fullbody Human Avatars" paper. This code is for the part of the paper describing video-based avatars.

NeuralTextures This is repository with inference code for paper "StylePeople: A Generative Model of Fullbody Human Avatars" (CVPR21). This code is for

Visual Understanding Lab @ Samsung AI Center Moscow 18 Oct 6, 2022
A code generator from ONNX to PyTorch code

onnx-pytorch Generating pytorch code from ONNX. Currently support onnx==1.9.0 and torch==1.8.1. Installation From PyPI pip install onnx-pytorch From

Wenhao Hu 94 Jan 6, 2023
This is the code for our KILT leaderboard submission to the T-REx and zsRE tasks. It includes code for training a DPR model then continuing training with RAG.

KGI (Knowledge Graph Induction) for slot filling This is the code for our KILT leaderboard submission to the T-REx and zsRE tasks. It includes code fo

International Business Machines 72 Jan 6, 2023
Convert Python 3 code to CUDA code.

Py2CUDA Convert python code to CUDA. Usage To convert a python file say named py_file.py to CUDA, run python generate_cuda.py --file py_file.py --arch

Yuval Rosen 3 Jul 14, 2021
Empirical Study of Transformers for Source Code & A Simple Approach for Handling Out-of-Vocabulary Identifiers in Deep Learning for Source Code

Transformers for variable misuse, function naming and code completion tasks The official PyTorch implementation of: Empirical Study of Transformers fo

Bayesian Methods Research Group 56 Nov 15, 2022
Reference implementation of code generation projects from Facebook AI Research. General toolkit to apply machine learning to code, from dataset creation to model training and evaluation. Comes with pretrained models.

This repository is a toolkit to do machine learning for programming languages. It implements tokenization, dataset preprocessing, model training and m

Facebook Research 408 Jan 1, 2023
Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

CoProtector Code for the prototype tool in our paper "CoProtector: Protect Open-Source Code against Unauthorized Training Usage with Data Poisoning".

Zhensu Sun 1 Oct 26, 2021
Low-code/No-code approach for deep learning inference on devices

EzEdgeAI A concept project that uses a low-code/no-code approach to implement deep learning inference on devices. It provides a componentized framewor

On-Device AI Co., Ltd. 7 Apr 5, 2022
Code for all the Advent of Code'21 challenges mostly written in python

Advent of Code 21 Code for all the Advent of Code'21 challenges mostly written in python. They are not necessarily the best or fastest solutions but j

null 4 May 26, 2022
Code to use Augmented Shapiro Wilks Stopping, as well as code for the paper "Statistically Signifigant Stopping of Neural Network Training"

This codebase is being actively maintained, please create and issue if you have issues using it Basics All data files are included under losses and ea

J K Terry 32 Nov 9, 2021