-
Notifications
You must be signed in to change notification settings - Fork 3.1k
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Added SiamMask CUDA implementation [2x faster on Full HD] #3571
Conversation
@@ -18,10 +18,20 @@ spec: | |||
|
|||
build: | |||
image: cvat/pth.foolwood.siammask | |||
baseImage: continuumio/miniconda3 | |||
baseImage: ubuntu:20.04 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Changed base image to have the same set of preCopy directives.
But finally it was nesessary to add
pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
to function-gpu.yml, so it is not the same now anyway. Let me know if you think that better to leave this file as is.
hi @bsekachev i am new to cvat, request you explain how to run SiamMask CUDA implementation from starting? |
Probably you use develop branch. This PR is not merged yet. To get function-gpu.yml you need to checkout to branch To deploy the function you either run |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM, but not able to test GPU version. And one question: is there option to autoupdate function with updating CVAT images?
Successfully built serverless function with GPU support under WSL, but eventually got an error when called a function. Probably it's a problem with my setup. Nuclio function log
|
@rohitsaluja22, also you can not rename a file and just add the flag |
@bsekachev , if annotations are removed using "Remove annotations", tracked objects always are reinitialized. |
Andrey, could your team update tracking documentation? |
@bsekachev, sure, we'll do it. |
I dont know why automatic annotation for "Allow Siammask active tracker to run for more frames" was closed, at the moment it can only run on the frame you are on and is incredibly slow |
General
Added CUDA support for SiamMask tracker
Tracker is able to track different objects in one server request
"Trackable" objects are introduced (an object will be tracked automatically if the previous frame was a latest keyframe for the object)
Added tracker indication on canvas
Added tracker switcher on sidebar
Tracking with non-blocking UI
Motivation and context
Resolved #2949
Resolved #2773
Related #2936
How has this been tested?
Checklist
develop
branchcvat-core, cvat-data and cvat-ui)
License
Feel free to contact the maintainers if that's a concern.