8af5800b9a74cf7b2c8d2577b7d9be3ef987f26b,detect.py,,detect_image,#Any#Any#Any#Any#Any#,65

Before Change

    Tensor = torch.cuda.FloatTensor if torch.cuda.is_available() else torch.FloatTensor

    // Configure input
    input_img = Variable(image.type(Tensor))

    // Get detections
    with torch.no_grad():

After Change

    model.eval()  // Set model to evaluation mode

    // Configure input
    input_img = transforms.Compose([
        DEFAULT_TRANSFORMS, 
        Resize(img_size)])(
            (image, np.zeros((1, 5))))[0].unsqueeze(0)

    if torch.cuda.is_available():
        input_img = input_img.to("cuda")

    // Get detections
    with torch.no_grad():
        detections = model(input_img)

Instances