Circle Detection in Noisy Images using CNN

This repository contains code for estimating circle parameters (i.e. radius and centre coordinates) in noisy images using a deep network.

Getting Started

Clone the repo:

git clone https://github.com/lukan94/noisy_circle_detection.git

Requirements

Tested with python 3.6

  • python packages
    • pytorch==0.4.1
    • numpy
    • opencv
  • NVIDIA GPU and CUDA 9.0

Data Preparation

Training

Inference