|
@@ -23,7 +23,6 @@ from libs.vision_libs.models.detection.backbone_utils import _resnet_fpn_extract
|
|
|
BackboneWithFPN, resnet_fpn_backbone
|
|
BackboneWithFPN, resnet_fpn_backbone
|
|
|
from libs.vision_libs.models.detection.faster_rcnn import FasterRCNN, TwoMLPHead, FastRCNNPredictor
|
|
from libs.vision_libs.models.detection.faster_rcnn import FasterRCNN, TwoMLPHead, FastRCNNPredictor
|
|
|
from .heads.arc_heads import ArcHeads, ArcPredictor
|
|
from .heads.arc_heads import ArcHeads, ArcPredictor
|
|
|
-from .heads.arc_unet import ArcUnet
|
|
|
|
|
from .heads.circle_heads import CircleHeads, CirclePredictor
|
|
from .heads.circle_heads import CircleHeads, CirclePredictor
|
|
|
from .heads.line_heads import LinePredictor
|
|
from .heads.line_heads import LinePredictor
|
|
|
from .heads.point_heads import PointHeads, PointPredictor
|
|
from .heads.point_heads import PointHeads, PointPredictor
|