`create_extension` was removed in pytorch-1.0

create_extension was supported by pytorch v0.4.1 only starting 1.0 pytorch the function was erased in favour of cpp extensions but it’s not a drop-in replacement

[GreenTeaHua]

torchvision.ops.nms(boxes, scores, iou_threshold) Performs non-maximum suppression (NMS) on the boxes according to their intersection-over-union (IoU).

NMS iteratively removes lower scoring boxes which have an
IoU greater than iou_threshold with another (higher scoring)
box.

Parameters
----------
boxes : Tensor[N, 4])  bounding boxes坐标. 格式：(x1, y1, x2, y2)
    boxes to perform NMS on. They
    are expected to be in (x1, y1, x2, y2) format
scores : Tensor[N]   bounding boxes得分
    scores for each one of the boxes
iou_threshold : float IoU过滤阈值
    discards all overlapping
    boxes with IoU < iou_threshold

Returns
-------
keep : Tensor
    int64 tensor with the indices
    of the elements that have been kept
    by NMS, sorted in decreasing order of scores
    NMS过滤后的bouding boxes索引（降序排列）
"""

[sermakarevich]

Yes, I missed that code for pytorch 1.0 is not in master.