Move TCN vectorization into the TCN package

PTG-Kitware · Oct 24, 2024 · c6ba9be · c6ba9be
1 parent d7304a3
commit c6ba9be
Show file tree

Hide file tree

Showing 5 changed files with 4 additions and 1,089 deletions.
diff --git a/angel_system/activity_classification/tcn_hpl/predict.py b/angel_system/activity_classification/tcn_hpl/predict.py
@@ -17,7 +17,7 @@
 from tcn_hpl.data.components.augmentations import NormalizePixelPts, NormalizeFromCenter
 from tcn_hpl.models.ptg_module import PTGLitModule
 
-from angel_system.activity_classification.utils import (
+from tcn_hpl.data.vectorize_classic import (
     tlbr_to_xywh,
     obj_det2d_set_to_feature,
 )
@@ -515,7 +515,6 @@ def debug_from_array_file() -> None:
     import numpy as np
     import torch
     from tqdm import tqdm
-    from tcn_hpl.data.components.augmentations import NormalizePixelPts
     from angel_system.tcn_hpl.predict import (
         load_module,
         predict,

diff --git a/angel_system/activity_classification/train_activity_classifier.py b/angel_system/activity_classification/train_activity_classifier.py
@@ -3,7 +3,6 @@
 import pickle
 import kwcoco
 import argparse
-import json
 
 import numpy as np
 import matplotlib.pyplot as plt
@@ -16,7 +15,7 @@
 
 from angel_system.data.common.load_data import sanitize_str
 
-from angel_system.activity_classification.utils import (
+from tcn_hpl.data.vectorize_classic import (
     obj_det2d_set_to_feature,
 )