How to Create a Custom Object Detection Model with TensorFlow

Hey guys welcome back, Ben again! Today we are continuing the project we left off on last time. This time we are learning to detect custom objects using TensorFlow's Object Detection API. This video goes over installing LabelImg for creating our labels, generating csv files, and creating TFrecord files. The next video in this series will be training our custom model and testing it.

Commands:

Useful Links:

[masihzarafshan]

I love how you uploaded the whole video and did not edit out you errors. It helped me a lot.

[antoniojunior]

This is great, it's hard to find a good tutorial for custom object detection :) very good work!! Thanks Ben!

[rafaellauxen]

Thank you, 3 hours to find your video, your video was the only that works for i, thank you again!

[mehmetcanmitil]

just subscribed, very helpful videos, thank you so much man ! You helped me a lot !

[lkfdlzc]

Commands should be:
 python generate_tfrecord.py --image_dir=images/test/
 python generate_tfrecord.py --image_dir=images/train/

[nicolebaldy]

Very clear video, and very helpful. Some (minor) suggestions - list clearly which versions you are using (I started with Tensorflow 2 because I started here and not your install vide and it didn't work, but everything worked fine when I used 1.15 -- I could have missed the warning though). Can you also upload your notes as a pdf? They are difficult to see in how you stream now. Plus it's nice to have a notes sheet. Also, maybe make your terminal fill at least half the screen (Windows/Super + left/right arrow) with at least 1-2 font sizes larger? Great video, though

[aditiparetkar]

Thank you so much for this helpful video! Really helped for my semester 6 mini project :)

[waitwhat]

if anyone's getting the error image not found or no such directory use this
python generate_tfrecord.py --image_dir=images/train
and similarly for test

[achillesknox]

Thank You Bro. That was very Helpful.👍👍 Ur a gr8 Teacher.😎😎

[uxlhbpp]

Traceback (most recent call last):
File "generate_tfrecord.py", line 26, in <module>
from object_detection.utils import dataset_util
ModuleNotFoundError: No module named 'object_detection'
should i

[zetghiu]

I cannot run the final code
File "generate_tfrecord.py", line 24, in <module>
from object_detection.utils import dataset_util
ModuleNotFoundError: No module named 'object_detection'

what should i do?

[varunpurohit]

getting some pandas error in generate_tfrecord.py...can nyone help or can you tell how to execute that generate_tfrecord.py file in google colab....
please can anyone help?

[raygud]

AttributeError: module 'tensorflow' has no attribute 'app' any one have a solution for this ? i tried downgrading tensorflow but it wouldent let me

[creativeinstincts]

Must be a lot of eagle eyes out there. I don't have magnifying glass big enough to see your tutorial

[yigitaltun]

Hello, I want to do this process not for a single category but for 2 different categories, for example, apple and pear, how can I do it with 2 categories?

[tegaogheneovo]

Ama just subscribe real quick cause you talking bout it..tech tech let's go

[abheeshtamn]

QMimeDatabase: Error loading internal MIME data
An error has been encountered at line 1 of <internal MIME data>: Premature end of document.:

[shaneabreu]

Does anyone know why I have an error of 'no module named object_detection' please? It's when I try to generate the TFRecords. I am following step by step and am using PyCharm. Thanks

[ian-xtjy]

generate_tfrecord needs the following updates to run with TensorFlow 2:
(Warning! I'm still having trouble getting the training to start, and I'm still unsure if the errors I got later on are still stemming from my fixed generate_tfrecord. All I know is that these changes will not generate an error upon running the generate_tfrecord file.)

tf.app has been removed, but you can just as easily, with minimal changes, use absl-py package instead.
pip install absl-py
then on generate_tfrecord file, do:
import absl
Now, where on the file it says tf.app.flags, simply replace with absl.flags
and on the last line of the file, where it says tf.app.run(), simply replace it with absl.app.run(main)

All right, now two last minor changes:
on line 88, where it says tf.python_io.TFRecordWriter(FLAGS, output_path), change it to:

tf.io.TFRecordWriter(FLAGS, output_path)

and on line 47, where it says simply change it to

You're all set!!

The TensorFlow team does excellent work making sure all changes from TF1 to TF2 are not scary at all!

[aiemporium]

Excellent work sir. Thanks for sharing.
Sir, can i know which research paper are you used for this project as reference?
Kindly let me know about it sir.