Spaces:

AndreasLH
/

Weak-Cube-RCNN

Sleeping

AndreasLH commited on May 10

Commit

d805ffe

1 Parent(s): 6cff2b6

segment anything import remove

Files changed (1) hide show

cubercnn/modeling/roi_heads/roi_heads.py CHANGED Viewed

@@ -1,8 +1,8 @@
 from detectron2.layers.nms import batched_nms
 from pytorch3d.ops.iou_box3d import box3d_overlap
 from ProposalNetwork.utils.plane import Plane_torch as Plane_torch
-from segment_anything.utils.transforms import ResizeLongestSide
-from cubercnn.data.generate_ground_segmentations import init_segmentation
 import logging

 from detectron2.layers.nms import batched_nms
 from pytorch3d.ops.iou_box3d import box3d_overlap
 from ProposalNetwork.utils.plane import Plane_torch as Plane_torch
+# from segment_anything.utils.transforms import ResizeLongestSide
+# from cubercnn.data.generate_ground_segmentations import init_segmentation
 import logging