Buscar um retângulo na imagem

Tarefa : Encontrar um disco rígido na foto, determinar seu ângulo e contornos

Problema: Nem sempre é possível encontrar o contorno correto do disco.

No código, eu transformo as imagens em escala de cinza, azul, encontro a diferença entre elas e determino os contornos, mas a solução não é precisa. Gostaria da sua ajuda.

  firstframe = cv2.imread(bg_im) # Imagem de fundo nframe_img = cv2.imread(frame) # Imagem com um disco ngray = cv2.cvtColor(frame_img, cv2.COLOR_BGR2GRAY) ngray = cv2.GaussianBlur(gray, (21, 21), 0) nfirstframe = cv2.cvtColor(firstframe, cv2.COLOR_BGR2GRAY) nfirstframe = cv2.GaussianBlur(firstframe, (21, 21), 0) nframeDelta = cv2.absdiff(firstframe, gray) nthresh = cv2.threshold(frameDelta, 25, 255, cv2.THRESH_BINARY)[1] n