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

What's the dataset format for this project? #14

Open
iVanlIsh opened this issue May 15, 2018 · 11 comments
Open

What's the dataset format for this project? #14

iVanlIsh opened this issue May 15, 2018 · 11 comments

Comments

@iVanlIsh
Copy link

iVanlIsh commented May 15, 2018

I've tried scripts in data folder but seems the links are expired.
Then I tried to download dataset from http://host.robots.ox.ac.uk/pascal/VOC/voc2007/, and could not successfully run the code, too.
Please help me.
(也尝试魔改输入数据处理的代码,还是失败了,萌新求助QAQQQ)

@xhzdeng
Copy link
Owner

xhzdeng commented May 15, 2018

Unlike PascalVOC use [xmin, ymin, xmax, ymax] to make annotations, we utilize [x1, y1, x2, y2, x3, y3, x4, y4] to represent a object instance. So the [bndbox] field needs to have the following content:

<bndbox>
		<x1>158</x1>
		<y1>128</y1>
		<x2>158</x2>
		<y2>181</y2>
		<x3>411</x3>
		<y3>181</y3>
		<x4>411</x4>
		<y4>128</y4>
</bndbox>

The others are the same with PascalVOC. ^_^

@iVanlIsh
Copy link
Author

iVanlIsh commented May 15, 2018

I've changed that, but when I ran the project, it told me:

Loading pretrained model weights from models/pretrain.caffemodel
Traceback (most recent call last):
  File "./tools/train_net.py", line 112, in <module>
    max_iters=args.max_iters)
  File "../lib/fast_rcnn/train.py", line 157, in train_net
    pretrained_model=pretrained_model)
  File "../lib/fast_rcnn/train.py", line 53, in __init__
    self.solver.net.layers[0].set_roidb(roidb)
  File "../lib/roi_data_layer/layer.py", line 69, in set_roidb
    self._shuffle_roidb_inds()
  File "../lib/roi_data_layer/layer.py", line 36, in _shuffle_roidb_inds
    inds = np.reshape(inds, (-1, 2))
  File "/usr/local/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 257, in reshape
    return _wrapfunc(a, 'reshape', newshape, order=order)
  File "/usr/local/lib/python2.7/dist-packages/numpy/core/fromnumeric.py", line 52, in _wrapfunc
    return getattr(obj, method)(*args, **kwds)
ValueError: cannot reshape array of size 5011 into shape (2)

Is it because I cannot just run it with original training/validation data of VOC 2007?

@PeiChenPing
Copy link

I met the same problem ,have you solved this? @iVanlIsh

@iVanlIsh
Copy link
Author

@PeiChenPing not yet

@xhzdeng
Copy link
Owner

xhzdeng commented May 15, 2018

@iVanlIsh @PeiChenPing The number of images must be even.

@iVanlIsh
Copy link
Author

iVanlIsh commented May 16, 2018

@xhzdeng

Solving...
F0516 16:19:09.485952 18831 syncedmem.cpp:71] Check failed: error == cudaSuccess (2 vs. 0)  out of memory
*** Check failure stack trace: ***
experiments/scripts/train.sh: line 59: 18831 Aborted                 (core dumped) ./tools/train_net.py --gpu 0 --solver models/${NET}/solver.pt --weights ${WEIGHTS} --imdb ${TRAIN_IMDB} --iters ${ITERS} --cfg models/${NET}/config.yml ${EXTRA_ARGS}

@xhzdeng
Copy link
Owner

xhzdeng commented May 16, 2018

@iVanlIsh As you see, it is out of memory, you need 8GB GPU memory at least for training.

@iVanlIsh
Copy link
Author

@xhzdeng But my GPU has 11177MiB.

@xhzdeng
Copy link
Owner

xhzdeng commented May 16, 2018

@iVanlIsh Is there some other programs occupied your GPU?

@iVanlIsh
Copy link
Author

No. @xhzdeng

@xhzdeng
Copy link
Owner

xhzdeng commented May 17, 2018

maybe this will help you. ^_^

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

3 participants