question-mark
Stuck on an issue?

Lightrun Answers was designed to reduce the constant googling that comes with debugging 3rd party libraries. It collects links to all the places you might be looking at while hunting down a tough bug.

And, if you’re still stuck at the end, we’re happy to hop on a call to see how we can help out.

How to perform faster-rcnn on new dataset?

See original GitHub issue

I have followed the instructions and managed to run the demo successfully. According to instructions, to train a model on another dataset, one should use a command like this: ./experiments/scripts/train_faster_rcnn.sh [GPU_ID] [DATASET] [NET]

where [DATASET] can be pascal or coco. How can I run it on another dataset where annotation files are in .txt format, whereas in pascal the annotations are in xml?

Issue Analytics

  • State:closed
  • Created 6 years ago
  • Comments:9

github_iconTop GitHub Comments

5reactions
kalaspuffarcommented, May 24, 2017

I’ve just done what you ask for. There is a few places you need to change the python code in order to accomplish that.

First of you need to copy the pascal_voc.py in lib/datasets directory and implement your own _load_pascal_annotation function.

Changing the classes your interested in and also where your data files are located could be done in the same file.

Next up you update the factory.py file to include your new dataset.

Lastly you need to update the ./experiments/scripts/train_faster_rcnn.sh and ./experiments/scripts/test_faster_rcnn.sh to include your dataset.

0reactions
lyltencentcommented, Sep 17, 2017

@kalaspuffar, Hi Daniel, thanks for your information.

Read more comments on GitHub >

github_iconTop Results From Across the Web

Understanding and Implementing Faster R-CNN: A Step-By ...
The Faster R-CNN model takes the following approach: The Image first passes through the backbone network to get an output feature map, and...
Read more >
Training Faster R-CNN Using TensorFlow's Object Detection ...
Step-by-step tutorial to train a faster R-CNN for object detection with TensorFlow using a custom dataset. Author(s): Buse Yaren Tekin ...
Read more >
How to train Faster R-CNN on my own custom dataset? #1028
2- use the pre-trained Faster RCNN on the VOC2007 as the initial weights to train it then on my own custom dataset. 3-...
Read more >
Train your own object detector with Faster-RCNN & PyTorch
In order to train an object detector with a deep neural network like Faster-RCNN we require a dataset. For this, I downloaded 20...
Read more >
Faster RCNN Python | Faster R-CNN For Object Detection
If possible, you can use a GPU to make the training phase faster. You can also try to reduce the number of epochs...
Read more >

github_iconTop Related Medium Post

No results found

github_iconTop Related StackOverflow Question

No results found

github_iconTroubleshoot Live Code

Lightrun enables developers to add logs, metrics and snapshots to live code - no restarts or redeploys required.
Start Free

github_iconTop Related Reddit Thread

No results found

github_iconTop Related Hackernoon Post

No results found

github_iconTop Related Tweet

No results found

github_iconTop Related Dev.to Post

No results found

github_iconTop Related Hashnode Post

No results found