Unknown builtin op: torchvision::nms

build.gradle

    plugins {
        id 'java'
    }
    
    repositories {
        mavenCentral()
    }
    
    dependencies {
        implementation group: 'ai.djl.pytorch', name: 'pytorch-engine', version: '0.10.0'
        implementation group: 'ai.djl.pytorch', name: 'pytorch-native-cpu', version: '1.7.1'
    
        implementation group: 'org.slf4j', name: 'slf4j-simple', version: '1.7.30'
    }

Predict.java

    import ai.djl.MalformedModelException;
    import ai.djl.Model;
    
    import javax.imageio.ImageIO;
    import java.awt.image.BufferedImage;
    import java.io.IOException;
    
    public class Predict {
        public static void main(String[] args) {
            Path modelDir = Paths.get("src/main/resources/fasterrcnn_resnet50.pt");
    
            try (Model model = Model.newInstance("fasterrcnn_resnet50")) {
                model.load(modelDir);
            } catch (IOException | MalformedModelException e) {
                e.printStackTrace();
            }
        }
    }

[stu1130]

@anandvsr nms is TorchVision operator that we don’t support. It is still blocked by https://github.com/pytorch/vision/issues/2692.

[smeet2804]

Thanks @stu1130 , I tried, but not able to run make command after cmake ... It gives make *** no targets specified and no makefile found. stop. Don’t know where I’m going wrong. I tried it on Windows 10 through anaconda prompt.