Using NMS for custom detections on Azure

[shreyydesai]

Hi @glenn-jocher,
I have been trying to deploy the model on Azure and I cannot deploy the entire yolov5 folder. Currently, it accepts image in the form of an array for inference. I have made changes to detect.py for the ensemble model. I do get outputs for the two models separately. However, It gives me nothing after NMS. Is there a way I could use some script that does the NMS after I get the results from the models (load the model from torch hub) and then deploy it to Azure. Thanks.