Skip to content
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

Open
Maeve-D opened this issue May 26, 2020 · 4 comments
Open

The result of training custom dataset #118

Maeve-D opened this issue May 26, 2020 · 4 comments

Comments

@Maeve-D
Copy link

Maeve-D commented May 26, 2020

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:
image

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

@ifzhang
Copy link
Owner

ifzhang commented May 28, 2020

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.

@Maeve-D
Copy link
Author

Maeve-D commented Jun 1, 2020

@ifzhang
I want to track airplane. I found some problem in the training dataset. Here I have two questions:
1.Do I need to label objects that are not moving but in the images?
2.My training images are obtained from the screenshot in the videos, and how many milliseconds is it appropriate to take a screenshot?
Thank you for your kindly help!: )

@NaifahNurya
Copy link

@Maeve-D Why don't you extract frame directly from the video instead of taking screenshots?

@geyu1998
Copy link

i use custom dataset to train,but there is an error: IsADirectoryError: [Errno 21] Is a directory: '/FairMOT-master/datas/'
can you solve this problem ,i put the custom dataset in datas file

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants