How can i detect and localize object using tensorflow and convolutional neural network?

user_5536013 0
Viewed 5927

My problem statement is as follows :

" Object Detection and Localization using Tensorflow and convolutional neural network "

What i did ?

I am done with the cat detection from images using tflearn library.I successfully trained a model using 25000 images of cats and its working fine with good accuracy.

Current Result : enter image description here

What i wanted to do?

If my image consist of two or more than two objects in the same image for example cat and dog together so my result should be 'cat and dog' and apart from this i have to find the exact location of these two objects on the image(bounding box)

I came across many high level libraries like darknet , SSD but not able to get the concept behind it.

Please guide me about the approach to solve the problem.

Note : I am using supervised learning techniques.

Expected Result :

enter image description here

tflearn tensorflow deep-learning computer-vision
3 Answers

I have done a similar project (detection + localization) on Indian Currencies using PyTorch and ResNet34. Following is the link of my kaggle notebook, hope you find it helpful. I have manually collected images from the internet and made bounding box around them and saved their annotation file (Pascal VOC) using "LabelImg" annotation tool.

https://www.kaggle.com/shweta2407/objectdetection-on-custom-dataset-resnet34

user_78501740
Related