Add yolov6 assigners to setup.py and remove extra paths

augmentedstartups · Sep 19, 2022 · d52af4f · d52af4f
1 parent 6d527e4
commit d52af4f
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -29,14 +29,12 @@ def setup_package():
                   f'{DISTNAME}.detectors',
                   f'{DISTNAME}.detectors.utils',
                   f'{DISTNAME}.detectors.temp',
-                  f'{DISTNAME}.detectors.temp.layers',
-                  f'{DISTNAME}.detectors.temp.models',
-                  f'{DISTNAME}.detectors.temp.utils',
                   f'{DISTNAME}.detectors.yolov5',
                   f'{DISTNAME}.detectors.yolov5.yolov5.models',
                   f'{DISTNAME}.detectors.yolov5.yolov5.utils',
                   f'{DISTNAME}.detectors.yolov6',
                   f'{DISTNAME}.detectors.yolov6.yolov6.layers',
+                  f'{DISTNAME}.detectors.yolov6.yolov6.assigners',
                   f'{DISTNAME}.detectors.yolov6.yolov6.models',
                   f'{DISTNAME}.detectors.yolov6.yolov6.utils',
                   f'{DISTNAME}.detectors.yolov7',