search

chenhaoxing / DiffusionInst

This repo is the code of paper "DiffusionInst: Diffusion Model for Instance Segmentation" (ICASSP'24).
Apache License 2.0
221 stars 12 forks source link

diffusioninst/__init__.py is empty #3

Closed enginst closed 1 year ago

enginst commented 1 year ago

Thank you for the work. "diffusioninst/init.py" is empty and can be filled with the following code.

from .config import add_diffusioninst_config
from .detector import DiffusionInst
from .dataset_mapper import DiffusionInstDatasetMapper
from .test_time_augmentation import DiffusionInstWithTTA
from .swintransformer import build_swintransformer_fpn_backbone
zhangxgu commented 1 year ago

@enginst Thanks for your advice! We have updated the file.