re-organize extension

Author: CharlesPikachu

.gitattributes

@@ -1,2 +1 @@
 * linguist-language=python
-

.gitignore

@@ -155,3 +155,4 @@ dmypy.json
 
 # Pyre type checker
 .pyre/
+

CITATION.cff

@@ -5,4 +5,4 @@ authors:
 title: "SSSegmentation: An Open Source Supervised Semantic Segmentation Toolbox Based on PyTorch"
 date-released: 2020-10-23
 url: "https://github.com/SegmentationBLWX/sssegmentation"
-license: Apache-2.0
+license: Apache-2.0

LICENSE

@@ -199,4 +199,3 @@
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
-

setup.py

@@ -91,11 +91,11 @@ def genpackagesitems():
 def getextensions():
     ext_modules = []
     if SSSEG_WITH_OPS:
-        srcs = ["ssseg/modules/models/segmentors/samv2/connected_components.cu"]
+        srcs = ["ssseg/modules/models/extensions/samv2/connected_components.cu"]
         compile_args = {
             'cxx': [], 'nvcc': ['-DCUDA_HAS_FP16=1', '-D__CUDA_NO_HALF_OPERATORS__', '-D__CUDA_NO_HALF_CONVERSIONS__', '-D__CUDA_NO_HALF2_OPERATORS__'],
         }
-        ext_modules.append(CUDAExtension('ssseg._C', srcs, extra_compile_args=compile_args))
+        ext_modules.append(CUDAExtension('ssseg.extensions.samv2._C', srcs, extra_compile_args=compile_args))
     return ext_modules
 
 

ssseg/modules/models/segmentors/samv2/connected_components.cu renamed to ssseg/modules/models/extensions/sam2/connected_components.cu

@@ -38,7 +38,7 @@ def getsdpasettings():
 
 '''getconnectedcomponents'''
 def getconnectedcomponents(mask):
-    from ssseg import _C
+    from ssseg.extensions.samv2 import _C
     return _C.get_connected_componnets(mask.to(torch.uint8).contiguous())