Comments (4)
Of course you can, if you have enough GPU memory.
from fpn_tensorflow.
@Kongsea But when I changed it, I got this error:
Traceback (most recent call last):
File "./tools/train.py", line 229, in
train()
File "./tools/train.py", line 38, in train
text=tf.shape(gtboxes_and_label_batch)[1])
File "./tools/../libs/box_utils/show_box_in_tensor.py", line 59, in draw_box_with_color
img_tensor = tf.squeeze(img_batch, 0)
File "/home/rf/anaconda2/lib/python2.7/site-packages/tensorflow/python/ops/array_ops.py", line 2630, in squeeze
return gen_array_ops.squeeze(input, axis, name)
File "/home/rf/anaconda2/lib/python2.7/site-packages/tensorflow/python/ops/gen_array_ops.py", line 7862, in squeeze
"Squeeze", input=input, squeeze_dims=axis, name=name)
File "/home/rf/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/op_def_library.py", line 787, in _apply_op_helper
op_def=op_def)
File "/home/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 3392, in create_op
op_def=op_def)
File "/home/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1734, in init
control_input_ops)
File "/home/anaconda2/lib/python2.7/site-packages/tensorflow/python/framework/ops.py", line 1570, in _create_c_op
raise ValueError(str(e))
ValueError: Can not squeeze dim[0], expected a dimension of 1, got 64 for 'draw_gtboxes/Squeeze' (op: 'Squeeze') with input shapes: [64,?,?,3].
What can I do? I just started doing this project so I need your help. Thanks a lot!
from fpn_tensorflow.
Sorry, it seems the original code only supports to set batch_size to be 1. Maybe you have to modify the code accordingly while changing the batch_size.
from fpn_tensorflow.
@Kongsea Thank you ! I'll try it.
from fpn_tensorflow.
Related Issues (18)
- OutOfRangeError: PaddingFIFOQueue '_2_get_batch/batch/padding_fifo_queue' is closed and has insufficient elements HOT 6
- 你好,看精度直接运行eval.py吗? HOT 5
- It seems stop when I train my data HOT 1
- test.py运行的图片是放在了哪里啊 HOT 1
- 为什么我训练时loss一直在上下波动并不收敛,结果也没有框出东西来,类别也是0,啥都检测不到
- 生成classes.txt出现问题 HOT 3
- 您好,请问fast-r-cnn的分类损失函数是否可以更换?我在换的过程中出现了.InvalidArgumentError: tags and values not the same shape: [] != [256] (tag 'RPN_LOSS/cls_loss')这个问题,请问您是否遇到过?
- 我的错误和 您提示得一样,可是按照修改提示修改之后还是会报错。
- 你使用的opencv是什么版本的啊? HOT 1
- 请问这个怎么调整batch_size
- do you have try this code on voc07+12? HOT 11
- SCALE_FACTORS = [10., 5., 1., 0.5]作用是什么。应该怎么修改 HOT 2
- TypeError: cond() got an unexpected keyword argument 'true_fn' HOT 2
- checkpoint_path error! HOT 1
- which datasets to use HOT 2
- Have you tried the VGg network? I have been trying to change the network, modify C2, C3, C4 and C5. There has been a mistake. Can you give me some advice? HOT 1
- How to use the pre-trained model?
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from fpn_tensorflow.