Тёмный

How to Train YOLOv5 on a Custom Dataset 

Roboflow
Подписаться 43 тыс.
Просмотров 221 тыс.
50% 1

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

 

30 сен 2024

Поделиться:

Ссылка:

Скачать:

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

Добавить в:

Мой плейлист
Посмотреть позже
Комментарии : 229   
@milesgiehtbrock8510
@milesgiehtbrock8510 4 года назад
Dude thanks for speaking in a way that teaches me instead of just assumes I already know this stuff
@Roboflow
@Roboflow 3 года назад
Certainly - thanks for tuning in.
@luqmansen
@luqmansen 4 года назад
I'm about to follow a yolo3 tutorial, and then I found the yolo4 tutorial, and then I found yolo5 tutorial, omg Thanks roboflow.
@Roboflow
@Roboflow 4 года назад
Thanks! Let us know what you want to see next.
@nadirjackson5755
@nadirjackson5755 3 года назад
When I try to run the train.py script I get the following issue: Traceback (most recent call last): File "train.py", line 492, in train(hyp, opt, device, tb_writer, wandb) File "train.py", line 91, in train model = Model(opt.cfg, ch=3, nc=nc).to(device) # create File "/content/yolov5/models/yolo.py", line 95, in __init__ self._initialize_biases() # only run once File "/content/yolov5/models/yolo.py", line 150, in _initialize_biases b[:, 4] += math.log(8 / (640 / s) ** 2) # obj (8 objects per 640 image) RuntimeError: a view of a leaf Variable that requires grad is being used in an in-place operation. Does anyone know How to Fix This?
@rezzailham7682
@rezzailham7682 2 года назад
can u explain what the meaning of "head", "backbone", and "anchors" on the models of yolov5?
@ammarfkhan
@ammarfkhan 4 года назад
from utils.utils import plot_results; plot_results() # plot results.txt as results.png Image(filename='./results.png', width=1000) # view results.png This is not working, I just copied your code, giving me an error "ModuleNotFoundError: No module named 'utils.utils'". Kindly anybody help me asap. Thanks.
@muskanrana5758
@muskanrana5758 3 года назад
18:11 - I don't get the question "replace data.yaml". any suggestions?
@tatianarice2259
@tatianarice2259 4 года назад
wow, great! thanks for sharing!
@yashrunwal5111
@yashrunwal5111 4 года назад
How do i test the model for a video? What should we change in the line 'detect.py'?
@Roboflow
@Roboflow 4 года назад
Note per the Ultralytics repo, when calling detect.py, source should be updated to "0" github.com/ultralytics/yolov5#inference
@snehitvaddi
@snehitvaddi 3 года назад
Will *weights file* and *config file* generated by Roboflow be acceptable by "dnn" module of OpenCV ie: cv2.dnn.readNetFromDarknet ??
@dinarakhaydarova4898
@dinarakhaydarova4898 2 года назад
perrrfectttt
@vendetta1273
@vendetta1273 6 месяцев назад
Can you tell how to change the model architecture? Like the layers. I would appreciate that. If you have already made a made for that please leave a link to that. Thanks.
@aissaelouahabi9503
@aissaelouahabi9503 2 года назад
do i need GPU?
@21_jtd3f_sarahputritaulada5
hi, may I ask how to use a "Try this model drop an image" box in a library that already exist? cuz when I drop the image it cant be detected an just shown this command { "predictions": [] }
@adityasareen6593
@adityasareen6593 7 месяцев назад
how can i use download .cfg and weights file to use it locally on my system using python and open cv code ??? plzz help
@amaltp4893
@amaltp4893 4 года назад
Its very nice. How can I download this model to my PC ? Is the weights are sufficient ?? I need the architecture also right ?? I want to make an android app for object detection , any tutorial available on that ? Pardon for any mistake in the question , I am a beginner.
@hackercop
@hackercop 3 года назад
Sir you are amazing thanks, you make we want to get roboflow Pro! Keep it up!
@pitiwatkittiwimonchai4656
@pitiwatkittiwimonchai4656 2 года назад
" define input image size " 19:47 I don't know what it mean ? can anyone help me explain what is number 416
@diegoG21
@diegoG21 3 года назад
sadly the collab notebook doesn't work anymore :c
@francescopaganelli177
@francescopaganelli177 3 года назад
Is it possible detect or recognize different person like different classes? Thanks great job!
@scott32714keiser
@scott32714keiser 3 года назад
Isn't there some offline way to do this I'm trying to build a robot that can learn like detect a human check the other dataset for name if no name pops up it start a low frame rate video recording of the human box and says hi my name is whatever what's your name what ever you say get processed and converted to text and repeats your name back to you and once you confirm your name it it says please turn around and let me look at you and 5 seconds later it says nice to meet you and by that time it's got enough video so that's stopped and converted into pictures frame by frame with the box already set from the human box a few raspberry pi's should do the trick one for the human box one for names one for body controls one for counting money I want to send it to the dollar store for me a guy trained it to see cards counting money should be easy all I need is the yolo working from there shell shript is like child's play to me I was coding since it-83+ came out in middle school but yolo is like made by some genius forgot about the normals with less than than a 140iq I'm not all that dumb iq is 111 but this shit makes me feel like maybe I should retire and just stick to I.T.
@aflifatma6960
@aflifatma6960 4 года назад
this error cat: /content/yolov5/models/yolov5s.yaml: No such file or directory
@tahamansoor7396
@tahamansoor7396 3 года назад
same did u find the workaround
@yurikleb
@yurikleb 4 года назад
min 18:20 is bit confusing, the command downloads the files into /content, but you open the /content/yolov5 folder? where do the files should go?
@tahamansoor7396
@tahamansoor7396 3 года назад
exactly did find the solution
@hypercloud9689
@hypercloud9689 Год назад
In very confused, once you mount the weights to your drive, how can you use it in your code????
@abinayakalyanaraman5982
@abinayakalyanaraman5982 6 месяцев назад
Can you please tell me about how you labelled your dataset?
@YasmineBenAbdelkader
@YasmineBenAbdelkader Месяц назад
amazing work thank you so much !
@dieweltentdecker5878
@dieweltentdecker5878 4 года назад
How can I use the weights?
@Roboflow
@Roboflow 4 года назад
You can download the weights (as our notebook does at its conclusion) and reuse them elsewhere.
@dieweltentdecker5878
@dieweltentdecker5878 4 года назад
Roboflow How can I convert it to weights?
@russf
@russf 4 года назад
epoch: 'epics' -> 'ee-pox'
@faishalichal8934
@faishalichal8934 2 года назад
Should we use CUDA for training??
@andywong5193
@andywong5193 2 года назад
How canI save the model and use it? So I don’t need to train again,and then use the model
@huyle44346
@huyle44346 5 месяцев назад
How to connect plc with roboflow software?
@mdtokitahmid2970
@mdtokitahmid2970 4 года назад
Hey! One qus. I have my own images and xml files given and downloaded in my pc. How should I import those to collab?
@hinainam5037
@hinainam5037 2 года назад
Hello sir can you please tell me how to find accuracy???
@Bushra_bushra99
@Bushra_bushra99 3 года назад
You don't show thé results of vali images ?
@sainandhan1108
@sainandhan1108 Год назад
Thanks a ton, I finished my project easily by referring this video
@Roboflow
@Roboflow Год назад
We love to hear that! What was the project that you worked on? :)
@sainandhan1108
@sainandhan1108 Год назад
@@Roboflow UAV Detection in Real Time
@Roboflow
@Roboflow Год назад
@@sainandhan1108 sweet! Is that something you can share? Seems very interesting
@harivarma3187
@harivarma3187 Год назад
@Sai Nandhan. Bro koncham adhi ravadam ledhu bro, ninnu contact avacha?
@sanabatool5573
@sanabatool5573 4 года назад
Please tell me how these saved weights can be used for prediction
@sanabatool5573
@sanabatool5573 4 года назад
How can i use this saved weights for future prediction
@iCore7Gaming
@iCore7Gaming 2 года назад
I have my own GPU i can train on, how would i go about doing that? don't want to use external websites with the datasets i have, as they are very limited.
@Yazlm-hu3zs
@Yazlm-hu3zs 7 месяцев назад
How can I use this best.pt file?
@fakrulislam3140
@fakrulislam3140 4 года назад
# this is the YAML file Roboflow wrote for us that we're loading into this notebook with our data %cat data.yaml I do not see the file and get the "error cat: data.yaml: No such file or directory"
@tranmani
@tranmani 4 года назад
same problem
@delibrain2216
@delibrain2216 3 года назад
Hi Joseph .A great tutorial. The dataset you used was heavily unbalanced. The augmentation is performed on all images combined and not the ones that were misrepresented(WBC AND platelets). Don't you thing that the trained model will be biased towards RBC? Could you give some suggestions how to cope with unbalanced dataset? or should I perform augmentation on the least represented classes?
@philippehuet8347
@philippehuet8347 4 года назад
Great tutorial, maybe the best I have seen on Yolo. Thanks for sharing the Colab notebook
@Leemarkkk
@Leemarkkk Год назад
where do we get the code snippet??
@HifzaTariq-u7i
@HifzaTariq-u7i Месяц назад
SIR YOU R TOO Fast
@hemanthreddy8991
@hemanthreddy8991 4 года назад
Hey can any one say how to pass an image which in my desktop directory to detect and I have one class only
@tylersnard
@tylersnard 4 года назад
This worked GREAT for me! The best part was that it showed my training images and their labels, and it made me realize my training set was mis-labeled :)
@Roboflow
@Roboflow 4 года назад
Great to hear! If you're interested, we'd be happy to share your success on our blog. Write us in-app or here: docs.roboflow.ai/support
@tylersnard
@tylersnard 4 года назад
@@Roboflow Great, I just did!
@raunakmitra7868
@raunakmitra7868 Год назад
Brilliantly done! You know how to keep the viewers glued to your video.
@salma5772
@salma5772 Месяц назад
i have watched like above 20 videos for this purpose but i mean it this one vid is much more simplified easy steps going on just well explained good quality, not too fast not too slow or boring explanation just amazing !
@nurhanisahazmi9636
@nurhanisahazmi9636 3 года назад
I encountered some problem while executing this line: %%time %cd /content/yolov5/ !python train.py --img 416 --batch 16 --epochs 100 --data '../data.yaml' --cfg ./models/custom_yolov5s.yaml --weights '' --name yolov5s_results --nosave --cache and it comes out like this: /content/yolov5 Traceback (most recent call last): File "train.py", line 481, in opt.data, opt.cfg, opt.hyp = check_file(opt.data), check_file(opt.cfg), check_file(opt.hyp) # check files File "/content/yolov5/utils/general.py", line 70, in check_file assert len(files), 'File Not Found: %s' % file # assert file was found AssertionError: File Not Found: ./models/custom_yolov5s.yaml CPU times: user 8.98 ms, sys: 6.35 ms, total: 15.3 ms Wall time: 1.92 s I have no idea what went wrong, can you help me :(
@tahamansoor7396
@tahamansoor7396 3 года назад
same
@tahamansoor7396
@tahamansoor7396 3 года назад
@Sulastri Tambunan M1141447 yes its just a file path issue.
@rozhinataghdimi335
@rozhinataghdimi335 3 года назад
@@tahamansoor7396 how did you fix it?
@tahamansoor7396
@tahamansoor7396 3 года назад
check the model path.
@alperenkuzhan1407
@alperenkuzhan1407 10 месяцев назад
Hi, I'm working on a project and I need help with something. How do I find the correct epoch size for my project? Different tutorial videos on your channel have different epoch sizes. How do you determine this?
@TC-td7wk
@TC-td7wk 2 года назад
Hi, my folder 'weights' is not able to get the 'last_yolov5s_results.pt', the unique file contained after running all the code is 'download_weights.sh'. Does anyone know how to fix this issue? Thanks so much in advance.
@sahilapte7735
@sahilapte7735 Год назад
hello anyone there to help me>?
@nunobartolo2908
@nunobartolo2908 3 года назад
how many trainable parameters does yolo5 have? does your training start with a pre trained model that used the coco dataset? if you use existing weights won't you need to re code the network head to change the classes to your own?
@JuneLetsPlays
@JuneLetsPlays 3 года назад
Trying to do this but getting the following error: Traceback (most recent call last): File "train.py", line 531, in train(hyp, opt, device, tb_writer, wandb) File "train.py", line 191, in train assert mlc < nc, 'Label class %g exceeds nc=%g in %s. Possible class labels are 0-%g' % (mlc, nc, opt.data, nc - 1) AssertionError: Label class 1 exceeds nc=1 in ../data.yaml. Possible class labels are 0-0
@antonioconsiglio95
@antonioconsiglio95 2 года назад
Really boring and unuseful
@kayalalakshmiswetha9527
@kayalalakshmiswetha9527 Год назад
I am not getting test train val folders in colab after entering the key in colab.. Can anyone help me with this...and I am also unable to get testvalues file for generating ground images...I need help
@Roboflow
@Roboflow Год назад
Hi 👋🏻 could you create new topic here: github.com/roboflow/notebooks/discussions/categories/q-a Describe your problem. I’ll try to help you.
@gguy156
@gguy156 2 года назад
I honestly don't understand why this model is working in Colab and does not working on my PC. It's always displaying nan loss. But pytorch setup is working well with other network architectures. What can be a problem with it?
@ThangNguyen-dh1dk
@ThangNguyen-dh1dk 3 года назад
how to show confusion matrix
@abdullateefagbaje1032
@abdullateefagbaje1032 2 года назад
This is a great video. I have a question, what if I want to use TensorFlow instead of PyTorch for YOLOv5 or 4?
@sstainba
@sstainba 2 года назад
I've been trying to use Yolov5 with PyTorch for the past few days... but there are some massive memory leaks during training. I'm constantly having to reboot.
@A_tavv
@A_tavv 4 месяца назад
this was the best video in the detection area by YOLO, believe me I tried a bunch of videos, non of them worked like this Thanks truly.
@xuxiang9529
@xuxiang9529 3 года назад
I following all steps you provided, but there is no predictions on the test dataset. I don't know why!!!!
@Clash_Royale_Iranian
@Clash_Royale_Iranian 2 года назад
Hi Joseph Is it possible to save the trained model, instead of weights? like TensorFlow "model.save" ??
@Readwithaami
@Readwithaami 2 года назад
I am a student .. I have used free plan roboflow for annotating data.but I couldn't download the dataset .. Export was not there ..so what should I do
@truptimahendrakar3486
@truptimahendrakar3486 3 года назад
@Roboflow "RuntimeError: non-positive stride is not supported" I get this error when I run the training command. My architecture is the same as the video. Does anyone else get this?
@shahrinnakkhatra2857
@shahrinnakkhatra2857 3 года назад
How to use this on video?
@Team-fh2rr
@Team-fh2rr Год назад
Good day! We have some issues with object detection on custom datasets and would be appreciated any advice. Is it possible to have a 1-hours consultation?
@itsshort6121
@itsshort6121 2 года назад
I having issue with my dataset. my dataset is video based and it anootation file for bounding box is also given how can I convert it for yolo v5
@QuickBrain
@QuickBrain 3 года назад
Could you please share how to run yolov5 trained using python on iOS and Android using coreml or pytorch mobile?
@hamzehshehab6167
@hamzehshehab6167 2 года назад
How can it be used to detect a person running in a stadium i mean how can i use yolov5 in activity recognition?
@ChafiaIbtissamkl
@ChafiaIbtissamkl 3 года назад
Dude thanks this video was really helpful. Just asking when i wanted to use roboflow to create my own data i didn't found the place where i have to use to generate data is it the new version that's why. Btw thanks again.
@andrewp494
@andrewp494 3 года назад
How to change the bounding box thickness and label ?I mean if you have multible object you can't see the details in the img
@will_tha_thrill
@will_tha_thrill 4 года назад
first
@dianavelychko6703
@dianavelychko6703 5 месяцев назад
this video is so transparent and really easy to understand and follow
@medcherifbraham1132
@medcherifbraham1132 3 года назад
thanks for the tutorial, but when i train my data it takes up to 5mins to complete one epochs on google colab, any help please ?
@srushti9694
@srushti9694 2 года назад
Please upload video of auto annotation for custom dataset using roboflow
@Neha-rh4gs
@Neha-rh4gs 2 года назад
how to label images
@prakulsingla7112
@prakulsingla7112 2 года назад
i have a json file and a zip file of yolo format. it contains txt files of all the images. which one should I use
@moahaimen
@moahaimen 2 года назад
i need a full explanation of tensor board measures and understand them please
@nicolassy1927
@nicolassy1927 3 года назад
Is it possible to just augment to the objects that yolov5 can detect already? Like for example I have a dataset of streetview images and I want to detect trees aside from cars and motorcycles. Do I have to retrain yolov5 again to detect cars and motorcycles, when it is already included in the original coco128.yaml file?
@PP-fh2mk
@PP-fh2mk 3 года назад
At 18:35 1) If we want to combine our custom set in default which yolov5 given sets is it possible? and how 2) or can we delete some classes that we doesn't needs? I would like to make custom sets with person classes, But if I made custom sets in your videos' method Maybe i think i have to make person dataset again . ( I'd like to add my customs to the existing default datasets. )
@raygao9954
@raygao9954 3 года назад
How do i convert the .pt file to .tflite? I have much difficulty doing that
@MYance-ut3ud
@MYance-ut3ud 4 года назад
how could i run this in my Ubuntu Desktop? its just download all those files? and then witch .py should i run?
@jagannathbehera9483
@jagannathbehera9483 2 года назад
Thanks Roboflow and Joseph for wonderful tutorial, but im struggling to convert best.pt to .onnx file as all tutorials using .pth extension and torch.load() failing for .pt file, please help me on this
@Kishi1969
@Kishi1969 3 года назад
Dear sir.. where is the video you used in the lesson located sir?
@tahaakgungor1026
@tahaakgungor1026 3 года назад
thank you for video and ı want to ask how can I find this trained model on mine local gpu on pycharm or jupyter
@suryasingh425
@suryasingh425 2 года назад
How can we count number of occurrence of each class in a video?
@aysebeyzaunal3472
@aysebeyzaunal3472 3 года назад
how can i download my trained dataset in my computer?
@MrTonmoy69
@MrTonmoy69 3 года назад
i got error when i tried to fit my dataset into yolo5 or yolo4 model.
@mayamehta2220
@mayamehta2220 2 года назад
Thank you Roboflow , thank you Joseph for this Excellent tutorial and colab notebook. I want to create custom dataset for my final year project. Can you please guide what should be image size and how to create dataset? my dataset contains different types of invoices
@martinolmos1557
@martinolmos1557 3 года назад
Thanks for sharing, como dijeran en mi México lindo, eres la v3rga!!!!
@zombiefreak5780
@zombiefreak5780 Год назад
Hello, do you still provide assistance for labelling custom dataset? I would like some guide on how to annotate on roboflow and export that annotated dataset to train my yolov7 model, but I am getting errors I hope you respond :( thank you
@Roboflow
@Roboflow Год назад
Hello! We have this guide for YOLOv7, hope it helps! blog.roboflow.com/yolov7-custom-dataset-training-tutorial/
@zombiefreak5780
@zombiefreak5780 Год назад
@@Roboflow hi! thanks so much for replying, really appreciate it, but may I ask if I could use my system graphics card while doing it on google colab, and if incase I would like to do commands on my windows11 cmd prompt, would the command be the same. I have been able to train and detect on the cmd prompt but now Im having a hard time trying to figure out how to add more to my already trained yolov7 model..i believe this is what you call freeze? sorry im new to coding and stuff. I would highly appreciate your help. thanks
@bbb-th1vu
@bbb-th1vu 3 года назад
i love your facial expressions. thanks. i dont know anything yet but hopefully i am planning to go into this, this summer.
@Arhan3l
@Arhan3l 4 года назад
How can I use Yolov5 to track objects? I see videos with Yolov3 + deep sort but don't know if I can just swap Yolo versions there.
@sarahch8878
@sarahch8878 3 года назад
Hello, did you have an answer to your question ? same here, I want to use YoloV5 to track custom objects using deepSORT..
@Arhan3l
@Arhan3l 3 года назад
@@sarahch8878 Unfortunately I don't have an answer to that question. I remember ending up using Yolov4 and finished the project. Maybe something changed, but taking into account your question It seems like it didn't.
@sarahch8878
@sarahch8878 3 года назад
@@Arhan3l thank you for your answer, may be I'll try to add some code from yolov3+deepSort, but it not seems to be easy... Did you find yolov4 with deepSort ? thank you
@Arhan3l
@Arhan3l 3 года назад
@@sarahch8878 Yes, there should be a video on youtube on the "The AI Guy" channel If I remember correctly. I was also searching some forums and the "Computer Vision" subreddit.
@sarahch8878
@sarahch8878 3 года назад
@@Arhan3l Thank you
@yatharthbhargava4257
@yatharthbhargava4257 4 года назад
how can I run the inference on my webcam?
@Roboflow
@Roboflow 4 года назад
When you run detect.py, specify source as "0." See more here: github.com/roboflow-ai/yolov5
@virajshirsekar4336
@virajshirsekar4336 4 года назад
@@Roboflow NOt able to run it on Webcam , getting error VIDEOIO ERROR: V4L: can't open camera by index 0
@namangupta1249
@namangupta1249 3 года назад
Hello sir, I have followed the same steps you have Given to train the model, but my MAP is decreasing per epoch. Please let me know how to solve this. Thanks in Advance
@christiaanboerkamp1043
@christiaanboerkamp1043 3 года назад
When I run the program it cannot find the yaml file do you know what's going on?
@JackJGX
@JackJGX 4 года назад
Roboflow is pretty awesome
@Roboflow
@Roboflow 4 года назад
Thank you!
@backdoorcynet1344
@backdoorcynet1344 3 года назад
Hi, it is a great project. Can yolov5 work on mobile devices or run live webcam on google colab?
@salmashahid7612
@salmashahid7612 3 года назад
Can anyone guide me on how can I extract the predicted boundary box image and save that image in a folder? I used the same Yolo v5 custom dataset training strategy.
@victor22227
@victor22227 4 года назад
Hey Nice video! Is there a way to like export the weights and such to use in a custom python project? I already trained a custom dataset, now I need to combine it with my python code that counts cars as they pass by. There seems to be no videos explaining how to use yolov5 other than on google collab
Далее
МОЮ ТАЧКУ РАЗБИЛИ...!
39:06
Просмотров 366 тыс.
DAXSHAT!!! Avaz Oxun sahnada yeg'lab yubordi
10:46
Просмотров 489 тыс.
YOLOv5 training with custom data
20:29
Просмотров 425 тыс.
МОЮ ТАЧКУ РАЗБИЛИ...!
39:06
Просмотров 366 тыс.