Merge pull request #166 from alexander-lynch/dev/novelty_detection

Dev/novelty detection
VIAME · Sep 24, 2023 · 8937478 · 8937478
2 parents 401e088 + 0e4801f
commit 8937478
Show file tree

Hide file tree

Showing 4 changed files with 161 additions and 1 deletion.
diff --git a/plugins/pytorch/CMakeLists.txt b/plugins/pytorch/CMakeLists.txt
@@ -89,9 +89,10 @@ if( VIAME_ENABLE_LEARN )
       arrows/pytorch
       remax_dino_detector )
 
-    kwiver_add_python_module(
+  kwiver_add_python_module(
       ${CMAKE_CURRENT_SOURCE_DIR}/remax_dino_trainer.py
       arrows/pytorch
       remax_dino_trainer )
 
+
 endif()
diff --git a/plugins/pytorch/remax/CMakeLists.txt b/plugins/pytorch/remax/CMakeLists.txt
@@ -0,0 +1,6 @@
+###
+# Loadable algorithms
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/remax/ReMax.py
+      arrows/pytorch/remax
+      ReMax)
diff --git a/plugins/pytorch/remax/model/CMakeLists.txt b/plugins/pytorch/remax/model/CMakeLists.txt
@@ -0,0 +1,59 @@
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/attention.py
+    arrows/pytorch/remax/model
+    attention)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/backbone.py
+    arrows/pytorch/remax/model
+    backbone)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/convnext.py
+    arrows/pytorch/remax/model
+    convnext)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/deformable_tranformer.py
+    arrows/pytorch/remax/model
+    deformable_tranformer)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/dino.py
+    arrows/pytorch/remax/model
+    dino)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/dn_components.py
+    arrows/pytorch/remax/model
+    dn_components)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/matcher.py
+    arrows/pytorch/remax/model
+    matcher)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/position_encoding.py
+    arrows/pytorch/remax/model
+    position_encoding)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/segmentation.py
+    arrows/pytorch/remax/model
+    segmentation)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/swin_transformer.py
+    arrows/pytorch/remax/model
+    swin_transformer)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/transformer_deformable.py
+    arrows/pytorch/remax/model
+    transformer_deformable)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/utils.py
+    arrows/pytorch/remax/model
+    utils)
diff --git a/plugins/pytorch/remax/util/CMakeLists.txt b/plugins/pytorch/remax/util/CMakeLists.txt
@@ -0,0 +1,94 @@
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/box_loss.py
+      arrows/pytorch/remax/util
+      box_loss)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/box_ops.py
+      arrows/pytorch/remax/util
+      box_ops)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/coco_eval.py
+      arrows/pytorch/remax/util
+      coco_eval)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/coco.py
+    arrows/pytorch/remax/util
+    coco)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/data_util.py
+    arrows/pytorch/remax/util
+    data_util)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/get_param_dicts.py
+    arrows/pytorch/remax/util
+    get_param_dicts)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/logger.py
+    arrows/pytorch/remax/util
+    logger)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/misc.py
+    arrows/pytorch/remax/util
+    misc)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/plot_utils.py
+    arrows/pytorch/remax/util
+    plot_utils)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/random_crop.py
+    arrows/pytorch/remax/util
+    random_crop)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/slconfig.py
+    arrows/pytorch/remax/util
+    slconfig)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/slio.py
+    arrows/pytorch/remax/util
+    slio)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/sltransform.py
+    arrows/pytorch/remax/util
+    sltransform)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/static_data_path.py
+    arrows/pytorch/remax/util
+    static_data_path)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/time_counter.py
+    arrows/pytorch/remax/util
+    time_counter)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/transforms.py
+    arrows/pytorch/remax/util
+    transforms)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/utils.py
+    arrows/pytorch/remax/util
+    utils)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/vis_utils.py
+    arrows/pytorch/remax/util
+    vis_utils)
+
+kwiver_add_python_module(
+    ${CMAKE_CURRENT_SOURCE_DIR}/visualizer.py
+    arrows/pytorch/remax/util
+    visualizer)