cnn_circle_detection

This is a PyTorch implementation of a Deep Convolutional Neural Network model for detecting circle objects in images.

Network Architecture

The output of the network is 4 real numbers which represents the detected center coordinates x,y,width and height.

Getting Started

Installation

• Clone this repo:

git clone https://github.com/wyattPol/circle_detection_cnn

• Requirements Install the dependencies by running the following command:

pip install -r requirements.txt

Model training

• Data set [here is dataset source]

https://universe.roboflow.com/trees-m4e49/-39p4p/dataset/1

• Training:

python train3.py

Model Testing

THere is already the test.py you can use(be careful with path):

python test.py

In this repo you can see 2 network, folder model_net is the trained model from network.py. folder model_net_1 contains the trained model from network2.py.

Results on examples

Display like below