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

Unable to convert Imagenet-VID to TFRecords for TF/lstm_object_detection #1

Open
wants to merge 4 commits into
base: master
Choose a base branch
from

Conversation

evgps
Copy link

@evgps evgps commented Sep 9, 2019

Dear @wakanda-ai and @KanaSukita I have found that you have succeeded in running code from tensorflow/models#6253 .
Can you please provide more instructions on how to use your convertor and which settings in https://github.com/tensorflow/models/blob/master/research/lstm_object_detection/configs/lstm_ssd_mobilenet_v1_imagenet.config are correct for usage with your code.

@KanaSukita
Copy link

Hi @evgps the command I use for the converter is:

python vid_2015_to_tfrecord.py --root_dir=/path/to/VID2015 --output_path=/path/to/output

It should generate tfrecords containing sequential data with length of 4, which is the same in the config. And in the config, you need to change the input path to your tfrecords.

BTW, optimization for MobilenetV2 is updated recently, you may try that. However, still I couldn't train a valid model using the newest config, so I doubt that the way I convert the tfrecord might be wrong.

@wangpichao
Copy link

@KanaSukita How to set the lab_map_path? Or what is it? How to generate it? Any help would be appreciated.
train_input_reader: {
shuffle_buffer_size: 32
queue_capacity: 12
prefetch_size: 12
min_after_dequeue: 4
label_map_path: "path/to/label_map"
external_input_reader {
[lstm_object_detection.protos.GoogleInputReader.google_input_reader] {
tf_record_video_input_reader: {
input_path: './research/lstm_object_detection/inputs/tfrecords/VID_2015-0.tfrecord'
data_type: TF_SEQUENCE_EXAMPLE
video_length: 4
}
}
}
}

@KanaSukita
Copy link

@wangpichao I write the vid_label_map.pbtxt myself, something like below:

item {
name: "n02691156"
id: 1
display_name: "airplane"
}
item {
name: "n02419796"
id: 2
display_name: "antelope"
}
item {
name: "n02131653"
id: 3
display_name: "bear"
}
...

@wangpichao
Copy link

@wangpichao I write the vid_label_map.pbtxt myself, something like below:

item {
name: "n02691156"
id: 1
display_name: "airplane"
}
item {
name: "n02419796"
id: 2
display_name: "antelope"
}
item {
name: "n02131653"
id: 3
display_name: "bear"
}
...

@KanaSukita thanks. Till now, are you still working on this project?

@KanaSukita
Copy link

@wangpichao No, I couldn't train a valid model even if I use the training data for evaluation. But I am willing to try again if there is any upate in the official repo.

@wangpichao
Copy link

@wangpichao No, I couldn't train a valid model even if I use the training data for evaluation. But I am willing to try again if there is any upate in the official repo.

@KanaSukita I also tried several times and with no valid models.

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

Successfully merging this pull request may close these issues.

3 participants