Train YOLOv10 on Custom Dataset

Подписаться 33 тыс.

Просмотров 8 тыс.

50% 1

Learn to perform custom object detection using YOLOv10.
GitHub: github.com/AarohiSingla/YOLOv...
Dataset: Dataset is also present in GitHub repo.
What is introduced in YOLOv10 : • Learn What Is Introduc...
Email: aarohisingla1987@gmail.com
YOLOv10: Real-Time End-to-End Object Detection
Paper: arxiv.org/pdf/2405.14458
YOLOv10, developed by researchers at Tsinghua University introduces a novel approach to real-time object detection. This version addresses deficiencies in both post-processing and model architecture found in earlier YOLO versions. By removing non-maximum suppression (NMS) and optimizing various model components, YOLOv10 achieves state-of-the-art performance with significantly reduced computational overhead. Extensive experiments show its superior accuracy-latency trade-offs across multiple model scales.
#computervision #objectdetection #yolov9 #yolov8 #yolov10

Наука

Опубликовано:

25 май 2024

Ссылка:

Скачать:

Готовим ссылку...

Добавить в:

Мой плейлист

Посмотреть позже

Комментарии : 78

@TekesteTesfay 25 дней назад

Thanks, I like your top-down approach from paper to code.

@CodeWithAarohi 25 дней назад

Thank you!

@WIRO13 Месяц назад

Hi, thanks for the excellent demostration!

@CodeWithAarohi Месяц назад

Glad you enjoyed it!

@abubakarsaleem5167 29 дней назад

How can we visualize the live training of YOLOv10, similar to how it is done for YOLOv5 using TensorBoard?

@icon_chatt 15 часов назад

hi can you please tell us from where you got the main pt file because i dont see it on their official site

@mh-ll 12 дней назад

Aarohi, this is a great video. Thank you.

@CodeWithAarohi 12 дней назад

Thank you!

@aiforeveryone Месяц назад

Amazing👍

@CodeWithAarohi Месяц назад

Thanks!

@yeongnamtan Месяц назад

Thank you for the clear explanation. Can you do one incorporating DeepSort tracking ?

@CodeWithAarohi Месяц назад

Will try!

@TomislavLevakovic Месяц назад

Great, as always ! Thank you ! I have one stupid question...Why Ultralytics says it is version ultralytics-8.1.34, i expect v10 will be V10?? Can you explain? Thanks !

@CodeWithAarohi Месяц назад

Because YOLOv10 is built on Ultralytics package. If you will open the official github repo of yolov10, you will see "ultralytics" folder there.

@TomislavLevakovic Месяц назад

@@CodeWithAarohi Thank you for answer, i understand that part. But, official package which you can install with pip, is on 8.2.23 version already. That is what confuses me. What is the difference between all those versions?? For example, why V10 cannot be installed on top of 8.2.23 version? Please, clafiry, beacuse for us, newcomers, it is very confusing all those versions and repos. Thank you !

@faizarehman2943 Месяц назад

Can you please guide me why other weights of YOLOV10 is not working on colab because small ansd nano are working.

@CodeWithAarohi Месяц назад

Not sure because haven't tried all the weights yet!

@faizarehman2943 Месяц назад

@@CodeWithAarohi can you please try other weights and guide me to solve my issue.

@user-jx2en8mo2b 5 дней назад

Hi. I have encountered an error that persists. I have tried everything like using absolute/relative paths, but the issue persists. Could someone help me with the following error? RuntimeError: Dataset 'data.yaml' error 'data.yaml' does not exist P.S. - I have checked the file format, its contents, and its path, everything seems all right, but the issue persists.

@AxelVinant 25 дней назад

Hello, thank you for the video. I'm actually using yolovn for my projects since years (ios and androids apps with live detection). I am currently a bit lost between all the new versions of yolo (v8, v9, v10...). Could you tell me what you think it is the best yolo for training and deploying models on ios and android ? Thank you for all your work !

@CodeWithAarohi 25 дней назад

My viewpoint: For mobile deployment, YOLOv8 is currently the best option due to its excellent balance of performance, ease of use, and robust community support. It works well with both TensorFlow Lite for Android and Core ML for iOS, making it suitable for real-time applications on mobile devices.

@AxelVinant 22 дня назад

@@CodeWithAarohi Thank you for your answser !

@g.s.3389 Месяц назад

where can I find a license plate dataset for Italy? or the one you used works anyway?

@CodeWithAarohi Месяц назад

The one I have used is present in github repo. Link shared in description section.

@g.s.3389 Месяц назад

@@CodeWithAarohi italian plates are different, will it work anyway?

@CodeWithAarohi Месяц назад

@@g.s.3389 if they are different then you need to prepare dataset according to that.

@raaginikalaimani217 16 дней назад

I tried to follow the tutorial but got stuck when i got module not found error, no module named “huggingface_hub” while i tried to predict “!yolo task=detect mode=predict conf=0.25 save=True, any comments on that?

@pifordtechnologiespvtltd5698 7 дней назад

Nice

@CodeWithAarohi 7 дней назад

Thanks

@jhicinternational Месяц назад

very good

@CodeWithAarohi Месяц назад

Thanks

@adibarnab8644 12 дней назад

Please make a video tutorial on Google Colab for YoloV10, Ma’am. It’s more convenient for us than Jupyter notebook.

@CodeWithAarohi 12 дней назад

Noted!

@jayathissaaral2598 Месяц назад

🥰🥰

@rickyS-D76 Месяц назад

Very nice explanation, thank you! I tried to follow the tutorial but got stuck when i got, "'yolo' is not recognized as an internal or external command, operable program or batch file." while i tried to predict, "!yolo task=detect mode=predict conf=0.25 save=True......".... any comments on that?

@CodeWithAarohi Месяц назад

github.com/ultralytics/ultralytics/issues/6281

@user-lx6uy8pc9b 28 дней назад

Amazing video. How to integrate yolov10 with SAM model ?

@CodeWithAarohi 28 дней назад

Will make a video

@user-vb8um5hp9s Месяц назад

isn't that a yolov8?

@jayathissaaral2598 Месяц назад

i have a little problem. why jupyter notebook all text display same color. this issue same as in my computer. do you know a way fix this ?

@nitinrai6093 Месяц назад

This may be because you may have converted the cell to markdown type, can you confirm if you can execute code on that cell? if not try converting it to code block again by 1. click on the cell > 2. hit esc button > 3. press M > 4. Press Y

@jayathissaaral2598 Месяц назад

@@nitinrai6093 no bro, in this video arohi selected code cell type. But still same as markdown cell type look

@DrWonderVoll 27 дней назад

Hi mam, Amazing work by the way. I am new to this object detection concept. I want to create a custom dataset with images of different fruits. How do I make the labels of those images? Thank you :)

@CodeWithAarohi 27 дней назад

Collect images of all types of fruits. Then use labelimg tool to label the images in yolo format which is txt format. check this: pypi.org/project/labelImg/

@DrWonderVoll 26 дней назад

@@CodeWithAarohi Thanks mam :)

@abubakarsaleem5167 28 дней назад

Hello ma'am, could you please explain that how we get the overall mAP of the test dataset if we tested the best.pt on our test dataset?

@siddhant8709 2 дня назад

Hi, aarohi how can you test the model with Inference on webcam

@CodeWithAarohi 2 дня назад

OK..Sure!

@abubakarsaleem5167 29 дней назад

Hello ma'am, could you please explain what 'train/box_om', 'train/cls_om', 'train/dfl_om', 'train/box_oo', 'train/cls_oo', and 'train/dfl_oo' mean in the plot after the YOLOv10 training is completed?

@CodeWithAarohi 28 дней назад

om is One-to-Many Head ( Generates multiple predictions per object during training to provide rich supervisory signals and improve learning accuracy) oo is One-to-One Head (Generates a single best prediction per object during inference to eliminate the need for NMS, thereby reducing latency and improving efficiency).

@abubakarsaleem5167 28 дней назад

@@CodeWithAarohi thanks Maam

@user-uq1qq7pj8h Месяц назад

Hello Madam, Thank you for your continuous support to students like me. I'm curious if you've implemented real-time YOLO image detection using a mobile phone's back camera from a browser before. Specifically, when a user opens the Flask web app and clicks a button, the back camera should automatically open and display live predictions. Currently, I can achieve this with a laptop webcam by setting the source to 0, but it doesn't work on a mobile phone. Do you have any sample projects or resources on how to do this? Your guidance means a lot.

@CodeWithAarohi Месяц назад

You can install IPcamera app on your phone and pass that ip camera address in source. I think this will work.

@user-uq1qq7pj8h Месяц назад

@@CodeWithAarohi I see, alright noted madam. Thanks a lot for your help Madam✨😇

@allanjobs3595 Месяц назад

Hi, I facing some error while I try to run this code in anaconda ! can you please share a video how to create a new env for that using GPU

@CodeWithAarohi Месяц назад

THis is how you can create environment: 1- Download and Install Python 2- Open your terminal (Command Prompt on Windows, Terminal on macOS/Linux). 3- Navigate to the directory where you want to create your project. 4- Then run these commands to create and activate environment: 5- python -m venv venv 6- venv\Scripts\activate

@m.abdullahgamer4494 27 дней назад

Can you please guide me how to setup virtual environment for yolov10 in anaconda. I am using jupyter notebook and made environment named yolov10 in anaconda. But yolov10 architecture gives error of yolo command not found.plz guide me

@CodeWithAarohi 27 дней назад

conda create -n yolov10 python=3.8 conda activate yolov10 conda install numpy pandas matplotlib conda install -c conda-forge opencv conda install -c conda-forge jupyterlab pip install torch torchvision torchaudio git clone github.com/ultralytics/yolov10.git cd yolov10 pip install -r requirements.txt

@m.abdullahgamer4494 26 дней назад

Thank you mam. I will try

@m.abdullahgamer4494 26 дней назад

Glad to implement yolov10 from the above advised steps by using environment. However ultralytics link is not working.Instead I clone from repository and it works

@m.abdullahgamer4494 26 дней назад

Your practical demonstration is highly appreciate able

@davidxu8352 Месяц назад

I am getting this error after finishing training the model when I want to predict (inference): i !yolo task=detect mode=predict conf=0.25 save=True model=runs/detect/train/weights/best.pt source=test_images_1/veh2.jpg 32 try: 33 # Issuing `None` to a generator fires it up 34 with ctx_factory(): ---> 35 response = gen.send(None) 37 while True: 38 try: ... --> 216 bs = prediction.shape[0] # batch size 217 nc = nc or (prediction.shape[1] - 4) # number of classes 218 nm = prediction.shape[1] - nc - 4 AttributeError: 'dict' object has no attribute 'shape' Output is truncated. View as a scrollable element or open in a text editor. Adjust cell output settings...

@CodeWithAarohi Месяц назад

add prediction = prediction['one2many'][0] to File "yolov10-main\ultralytics\utils\ops.py", line 216, in non_max_suppression bs = prediction.shape[0] # batch size final code: prediction = prediction['one2many'][0] bs = prediction.shape[0]

@arunmathur1668 День назад

Hi Aarohi, Can you please share the script to download the volov10 weight files that you had used in this video

@CodeWithAarohi День назад

You can download pretreained weights from here:github.com/THU-MIG/yolov10/releases

@nakulmali1413 24 дня назад

Hello mam first upon thanks for your all videos. Mam i request you please upload video on custom object classification using yolov8s . Model Training, validation and testing done with puthon script

@CodeWithAarohi 24 дня назад

Okay sure

@nakulmali1413 23 дня назад

Thanks mam

@harshays2873 Месяц назад

could you please make a video on yolo world custom dataset training

@CodeWithAarohi Месяц назад

For custom training of such models you need lot of data and variety of data which I don’t have to train custom yolo-world model. So, I can’t help you

@harshays2873 Месяц назад

@@CodeWithAarohi if possible could you please explain input data format so that we can try our own, it will be helpful, help us with initial training set up and data preparation.

@CodeWithAarohi Месяц назад

Check this page: docs.ultralytics.com/models/yolo-world/#set-prompts

@harshays2873 Месяц назад

@@CodeWithAarohi Thank you

@joserfjunior8940 4 дня назад

O minha delicia, da para fazer detectar a rua em quanto eu estiver dirigindo ? tipo um tesla copilot ? Otimo video minha delicia parabens !

@akhileshsharma5067 Месяц назад

Hello, I am not able to validate on test set, getting the following error: Traceback (most recent call last): File "/usr/local/bin/yolo", line 8, in sys.exit(entrypoint()) File "/usr/local/lib/python3.10/dist-packages/ultralytics/cfg/__init__.py", line 587, in entrypoint getattr(model, mode)(**overrides) # default args from model File "/usr/local/lib/python3.10/dist-packages/ultralytics/engine/model.py", line 518, in val validator(model=self.model) File "/usr/local/lib/python3.10/dist-packages/torch/utils/_contextlib.py", line 115, in decorate_context return func(*args, **kwargs) File "/usr/local/lib/python3.10/dist-packages/ultralytics/engine/validator.py", line 187, in __call__ preds = self.postprocess(preds) File "/usr/local/lib/python3.10/dist-packages/ultralytics/models/yolo/detect/val.py", line 86, in postprocess return ops.non_max_suppression( File "/usr/local/lib/python3.10/dist-packages/ultralytics/utils/ops.py", line 216, in non_max_suppression bs = prediction.shape[0] # batch size AttributeError: 'dict' object has no attribute 'shape'

@CodeWithAarohi Месяц назад

Add prediction = prediction['one2many'][0] to File "yolov10-main\yolov10-main\ultralytics\utils\ops.py", line 216, in non_max_suppression bs = prediction.shape[0] # batch size final code: prediction = prediction['one2many'][0] bs = prediction.shape[0]