-
Notifications
You must be signed in to change notification settings - Fork 935
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
The result of training custom dataset #118
Comments
May I ask what object you want to track? It seems the width is larger than the height. We filter the objects (only for person) when w is 1.6 times larger than h in track.py. You can check whether you have detection results. Also, '--conf_thres' can be adapted by your dataset. |
@ifzhang |
@Maeve-D Why don't you extract frame directly from the video instead of taking screenshots? |
i use custom dataset to train,but there is an error: IsADirectoryError: [Errno 21] Is a directory: '/FairMOT-master/datas/' |
I made custom dataset(5000 images ,target is airplane) and train the model followed the steps you mentioned in #55
The use:python:python train.py mot --exp_id all_dla34 --gpus 0 -batch_size 2 --load_model '../models/ctdet_coco_dla_2x.pth'
Here is the training dataset format:
00001.txt
0 1 0.076172 0.596875 0.150781 0.068750
0 2 0.601562 0.620833 0.095312 0.068750
0 3 0.967578 0.559896 0.064844 0.040625
0 4 0.651172 0.933854 0.253906 0.130208
The training process and loss value seem normal, here is the log.txt:
However, after training these I use: python demo.py mot --load_model ../exp/mot/all_dla34/model_last.pth --conf_thres 0.4 --input-video ../videos/myvideo.mp4 --output-root ../results/
But no was no any tracking in result videos.
I don't know which step is wrong. Can you give me some advise?
@ifzhang
The text was updated successfully, but these errors were encountered: