AssertionError: Selective search data not found at: tf-faster-rcnn/data/selective_search_data/voc_20

来源:互联网 发布:台湾服饰品牌知乎 编辑:程序博客网 时间:2024/05/17 01:40

I git stash and git pull,
then

rm -rf ./data/cache/./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16

Now it looks like this:

Preparing training data...
wrote gt roidb to /data/home/microway/experiments/tf-faster-rcnn/data/cache/voc_2007_test_gt_roidb.pkl
done
4952 validation roidb entries
Filtered 0 roidb entries: 10022 -> 10022
Filtered 0 roidb entries: 4952 -> 4952
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE3 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.1 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use SSE4.2 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use AVX2 instructions, but these are available on your machine and could speed up CPU computations.
W tensorflow/core/platform/cpu_feature_guard.cc:45] The TensorFlow library wasn't compiled to use FMA instructions, but these are available on your machine and could speed up CPU computations.
I tensorflow/core/common_runtime/gpu/gpu_device.cc:885] Found device 0 with properties:
name: Graphics Device
major: 6 minor: 1 memoryClockRate (GHz) 1.582
pciBusID 0000:02:00.0
Total memory: 10.91GiB
Free memory: 10.76GiB
I tensorflow/core/common_runtime/gpu/gpu_device.cc:906] DMA: 0
I tensorflow/core/common_runtime/gpu/gpu_device.cc:916] 0: Y
I tensorflow/core/common_runtime/gpu/gpu_device.cc:975] Creating TensorFlow device (/gpu:0) -> (device: 0, name: Graphics Device, pci bus id: 0000:02:00.0)
Solving...
Traceback (most recent call last):
File "./tools/trainval_net.py", line 131, in 
max_iters=args.max_iters)
File "/data/home/microway/experiments/tf-faster-rcnn/tools/../lib/model/train_val.py", line 378, in train_net
sw.train_model(sess, max_iters)
File "/data/home/microway/experiments/tf-faster-rcnn/tools/../lib/model/train_val.py", line 105, in train_model
anchor_ratios=cfg.ANCHOR_RATIOS)
File "/data/home/microway/experiments/tf-faster-rcnn/tools/../lib/nets/network.py", line 324, in create_architecture
val_summaries.append(self._add_image_summary(self._image, self._gt_boxes))
File "/data/home/microway/experiments/tf-faster-rcnn/tools/../lib/nets/network.py", line 56, in _add_image_summary
assert image.get_shape()[0] == 1
AssertionError



Change _C.TRAIN.PROPOSAL_METHOD = 'gt' the 118 line in model/config.py file. It should be OK.
We also met with this problem. Very weird.

阅读全文
0 0
原创粉丝点击