Skip to content

Conversation

@mattdawkins
Copy link
Member

…entifier

@BryonLewis
Copy link
Collaborator

Maybe we should do a review of all locations where we filter pipelines/training configs from VIAME?

Where pipelines are allowed/disallowed using RegEx within the Web version: https://github.com/Kitware/dive/blob/main/server/dive_tasks/pipeline_discovery.py#L20-L34

Where we get user-trained models, we filter by extension type here on the desktop:
https://github.com/Kitware/dive/blob/main/client/platform/desktop/backend/native/common.ts#L382

On the web for user trained models it looks like we are looking for [detector|tracker].pipe in their trained confg folders:
https://github.com/Kitware/dive/blob/main/server/dive_server/crud_rpc.py#L65
More specifically: https://github.com/Kitware/dive/blob/main/server/dive_server/crud_rpc.py#L98-105

Training Configs on the desktop web are specified here:
https://github.com/Kitware/dive/blob/main/client/platform/desktop/backend/native/common.ts#L453-L455

Training Configs on the web just seem to be filtered by '*.conf":
https://github.com/Kitware/dive/blob/main/server/dive_tasks/pipeline_discovery.py#L68

For Fine-Tuned Training on Desktop:
https://github.com/Kitware/dive/blob/main/client/platform/desktop/backend/native/common.ts#L455

This is looking inside of the user trained model folders for items for fine tune training:
https://github.com/Kitware/dive/blob/main/server/dive_server/crud_rpc.py#L128
These models are also added during the static training config import after using the CSV downloader for other models: https://github.com/Kitware/dive/blob/main/server/dive_tasks/pipeline_discovery.py#L97-L100
Training Extension matching uses this source:
https://github.com/Kitware/dive/blob/main/server/dive_utils/constants.py#L142

@BryonLewis BryonLewis self-requested a review September 15, 2025 17:04
@BryonLewis BryonLewis merged commit 296654b into Kitware:main Sep 17, 2025
4 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants