Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.
def censor(self, img_path, out_path=None, visualize=True, parts_to_blur=['BELLY', 'BUTTOCKS', 'F_BREAST', 'F_GENITALIA', 'M_GENETALIA', 'M_BREAST']):
if not out_path and not visualize:
print('No out_path passed and visualize is set to false. There is no point in running this function then.')
image = cv2.imread(img_path)
boxes = Detector.detect(self, img_path)
boxes = [i['box'] for i in boxes if i['label'] in parts_to_blur]
for box in boxes:
part = image[box[1]:box[3], box[0]:box[2]]
image = cv2.rectangle(image, (box[0], box[1]), (box[2], box[3]), (0, 0, 0), cv2.FILLED)
# image = cv2.GaussianBlur(part,(23, 23), 30)
# image[box[1]:box[3], box[0]:box[2]] = part
if visualize:
cv2.imshow("Blurred image", image)
cv2.waitKey(0)
if out_path:
cv2.imwrite(out_path, image)