frommmcv.opsimport(RoIAlign,SigmoidFocalLoss,nms,roi_align, 2+ sigmoid_focal_loss) 3+ 44 from.ball_queryimportball_query 55 from.furthest_point_sampleimportfurthest_point_sample 66 from.gather_pointsimportgather_points @@ -13,6 +13,7 @@ ...
import torch_mlu from mmcv.ops import sigmoid_focal_loss x = torch.randn(3, 10).mlu() x.requires_grad = True y = torch.tensor([1, 5, 3]).mlu() w = torch.ones(10).float().mlu() output = sigmoid_focal_loss(x, y, 2.0, 0.25, w, 'none') print(output) ``` 8 changes: ...
importnumpyasnpxv,yv=np.meshgrid(np.arange(3),np.arange(2))print(xv.flags['C_CONTIGUOUS'],yv.flags['C_CONTIGUOUS'])# True Trueimporttorchxv,yv=torch.meshgrid(torch.arange(3),torch.arange(2),indexing='xy')print(xv.is_contiguous(),yv.is_contiguous())# False Falsexv,yv=torch.meshgrid...
I try to install mmcv-full using below the version of cuda I have installed, some errors occur. But when I install pytorch using below the version of cuda I have installed, it also can work, do you have some idea to slove it? Maybe a suc...
mmcv-full OpenMMLab Computer Vision Foundation 15 nbclient A client library for executing notebooks. Formerly nbconvert's ExecutePreprocessor. 15 pylev A pure Python Levenshtein implementation that's not freaking GPL'd. 15 aesara A library for defining, optimizing, and efficiently evaluating mathematica...