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

可以训练,但是使用test_on_single_image.py测试训练好的模型没有bboxs,scores, #2

Open
pursuityu opened this issue Jan 18, 2020 · 10 comments

Comments

@pursuityu
Copy link

然后我又用官网提供的yolov3_coco.ckpt测试你的代码test_on_single_image.py,发现还是没有bbox,感觉是不是代码有问题,但是又没有报错。你测试有结果吗

@calmiLovesAI
Copy link
Owner

我测试的时候是正常的,因为使用test_on_single_image.py和在训练过程中开启测试单张图片用的是相同的代码。你在官网上下载的yolov3_coco.ckpt可能并不适用于这个项目,两者网络结构里面的细节不一定相同。

@herok94
Copy link

herok94 commented Jan 20, 2020

训练时候loss只有1.0左右,但是使用test_on_single_image.py没有输出,训练集有2400个样本,两类,只有正样本没有负样本

@pursuityu
Copy link
Author

训练时候loss只有1.0左右,但是使用test_on_single_image.py没有输出,训练集有2400个样本,两类,只有正样本没有负样本

我也不知道怎么回事,我换了代码。

@calmiLovesAI
Copy link
Owner

训练时候loss只有1.0左右,但是使用test_on_single_image.py没有输出,训练集有2400个样本,两类,只有正样本没有负样本

你可以尝试增加训练轮数

@PEPSIOFME
Copy link

我也训练完后预测没有输出结果,老哥很奇怪

@nanshenwei
Copy link

我也没有/(ㄒoㄒ)/~~

@nanshenwei
Copy link

我也没有/(ㄒoㄒ)/~~

再训练很多轮之后有结果了,不过结果不太好,换个图又没结果了,估计是需要训练再就一点 @pursuityu

@ZhouKai90
Copy link

我训练多轮之后,单张图片做测试,有框,但是是乱的,我发现bounding_box.py第23行附近
area = h * w
pred = tf.reshape(feature_map, shape=(-1, ANCHOR_NUM_EACH_SCALE * area, CATEGORY_NUM + 5))
pred = tf.nn.sigmoid(pred) #?????
tx_ty, tw_th, confidence, class_prob = tf.split(pred, num_or_size_splits=[2, 2, 1, CATEGORY_NUM], axis=-1)
这个地方对所有的结果都进行了sigmoid计算,我理解的是只用对其中的class_prob进行sigmod,而不应该对tx_ty, tw_th进行变换,是不是这里导致了坐标的不准确@calmisential

@holdrain
Copy link

调了调终于能训练了,不过缺点就是没有预训练的权重,emmmm

@holdrain
Copy link

用笔记本吃不消

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

7 participants