643bd0823c1f09fcb621207f3345879abf694ab0,demo.py,,main,#,85

Before Change

            // image = cv2.imread(image_list[image_idx])
            // image = utils.fixed_resize(image, (450, 450))
            dismap = utils.fixed_resize(dismap, (450, 450)).astype(np.uint8)
            cv2.imwrite("output.png", dismap)

            dismap = np.expand_dims(dismap, axis=-1)

After Change

            prediction[prediction<=0.8] = 0
            print("prediction done!")
            prediction = np.expand_dims(prediction, axis=-1).astype(np.uint8)
            image = image[:, :, ::-1] // change to bgr
            display_mask = np.concatenate([prediction, prediction, prediction], axis=-1)
            image = cv2.addWeighted(image, 0.9, display_mask, 0.5, 0.1)

Instances