Code Monkey home page Code Monkey logo

coarse_loftr_trt's Introduction

Hi there 👋

My name is Kirill Kolodiaznyi and I'm a software developer.

🔬 Research & Intersest

The research I’m working on now is the implementation of a robotic platform based on the NVIDIA Jetson Nano 2Gb device with monocular SLAM navigation. The main aim of this work is to improve feature matching algorithms used for camera positioning in a SLAM pipeline. The recent experiments are about using deep learning for SLAM, especially using the Local Feature Matching with Transformers for camera positioning.

Another topic of my interest is multi-threaded algorithms and using them for performance improvements in projects I’m working on. The result of this work is the library that represents higher-level, task-based parallelism. It’s based on the idea of the Intel TBB work-stealing threads management approach.

Also, I’m interested in learning and applying the C++ language for solving Machine Learning tasks. I wrote several articles and a book about the application of C++ to ML. Additionally, I made C++ implementations of popular object detection methods like Faster R-CNN and Mask R-CNN.

✍ Publications

🔧 Technologies & Tools

coarse_loftr_trt's People

Stargazers

 avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar  avatar

Watchers

 avatar  avatar  avatar  avatar  avatar

coarse_loftr_trt's Issues

only have coarse match ?

I tried webcam demo script, and found that only coarse match results are displayed, all points seems to lie on predefined grid, even with the original pretrained weights, predefined grid seem to be denser, matched points are not as good as the original LoFTR, do we miss any refine step in the eval script ?

Image Example

HI
Can u provide end-to-end example for image1<->image2 matching
thx

trt file running error

image
When I use the trt method to run webcam.py, I encounter the above error, how can I solve it

Output of onnx model

Hi author,
Your repo is veryhelpfull for my work, I have a question about ONNX model in your repo.
I find input of data in onnx model 2 image have shape 1,480,640
i dont understand output of model onnx(2 tensor with shape 1,1200,1200 ).
what output meanning. it is output of what step in LOFTR origin repo?
Thank you so much

problem about training

when i train the code ,the program feedback such information:
0%| | 0/1250 [00:05<?, ?it/s]
Traceback (most recent call last):
File "C:\Users\aust\Desktop\Coarse_LoFTR_TRT-main\train\trainer.py", line 154, in train_loop
self.scaler.scale(loss).backward()
File "C:\Users\aust\anaconda3\lib\site-packages\torch_tensor.py", line 255, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
File "C:\Users\aust\anaconda3\lib\site-packages\torch\autograd_init_.py", line 147, in backward
Variable._execution_engine.run_backward(
RuntimeError: Function 'KlDivBackward' returned nan values in its 0th output.
python-BaseException

it looks like the loss function has problem?

Accuracy tradeoff

Hi, does anyone now what the accuracy tradeoff is between the original loftr and the coarse loftr?

How to make my own dataset

Hello, I would like to ask how do I create my own dataset, what should be the size of the training samples, and how do I get the depth map of the image?

fine level matching

hi @Kolkir , nice work!
Could you explain why you didn't keep the fine level matching in this work? Is it difficult to achieve?

About the size of LoFTR_teacher.onnx

Hello,thank for your work!
I ran export_onnx.py, and the model(LoFTR_teacher.onnx ) size is 184MB, but the model size you provide is 1.76MB.
Could you fix it? Thanks a lot!

ASpanFormer,please try this!!

hi,professor:
a new image matching algorithm called ASpanFormer, ASpanFormer: Detector-Free Image Matching with Adaptive Span Transformer
github: https://github.com/apple/ml-aspanformer
this is amazing, please deploy this on nvidia jetson platform, the pipeline like: pytorch -> onnx -> tensorrt engine , then c++ deploy,
PLEASE!!!

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.