Repository navigation
Enable web image and video queries as isolated Jobs - #1994
Draft
mattdawkins wants to merge 1 commit into
Draft
mattdawkins wants to merge 1 commit into
mattdawkins wants to merge 1 commit into
Conversation
Collaborator
|
I'm not sure how well this workflow actuall works on web considering our deployed 2 GPU limit and the fact that those GPU's are almost always being used already. Because they are running long tasks (inference tracking, and training) you may have to wait hours to do your query. |
This branch has not been deployed
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
Adds image and video-frame similarity queries to DIVE Web. Users can build reusable private indexes, choose an uploaded image, an uploaded video frame, or a frame from an existing dataset, crop the exemplar, search, and refine results with positive/negative feedback. Results have cropped previews and links to the matching viewer frame. The last query and feedback marks restore after returning from the viewer.
All model execution runs as Jobs on the existing pipeline queue, including indexing, exemplar descriptor extraction, similarity search, and refinement. Private worker queues and the worker's GPU assignment are preserved. Saved SVM model export is deferred.
Execution and isolation
The browser only decodes media, captures exemplar frames, and renders crops. Client-side ONNX formulation is not introduced because this path preserves the existing descriptor/model contract.
Validation
Remaining integration validation
Draft pending an end-to-end run with an updated VIAME GPU worker. The local VIAME installation has the older PostgreSQL query pipeline, so GPU inference was not run outside Jobs.
On an updated deployment, build indexes and run image/video-frame searches and multiple refinement rounds while another user runs pipeline jobs. Verify cancellation, result frame alignment, returning from the viewer, and revoked dataset access. Workers need
configs/index.py, the file-backed query pipelines,viame.core.query_service, and their descriptor models.Text queries, saved SVM models, and saving query results as annotations remain desktop-only in this change. Index snapshots and query inputs/results are private Girder artifacts; unused artifacts can be removed through the Data browser. See
docs/Query.mdfor the web workflow and limits.