tensorflow-yolov3
tensorflow-yolov3 copied to clipboard
YunYang1994
自行用自己資料及訓練問題
自行用自己的資料庫訓練，訓練完有產生yolov3_test_loss=17.2442.ckpt-30這種meta檔案，之後還要跑convert_weight.py和freeze_weight.py這兩個檔案，但在跑convert_weight.py時出現runtimeerror的錯誤，想問在檔案裡有甚麼地方需要修改的，謝謝。
感謝tradysun的回覆，想問如果我checkpoint資料夾裡有這些檔案，那freeze_weight.py裡的ckpt要改成甚麼，謝謝。
現在想要做測試，想問在evaluate.py和main.py裡需要改甚麼嗎，跑完evaluate.py後產生下面第一個圖，再跑main.py後產生下面第二個圖，有錯誤，謝謝
你算出的mAP是0.00%,说明没有预测出任何结果，从你图1的log结果可看出predict为空，应该是你训练的结果有问题，是不是用的ckpt loss太大，模型还没收敛。
另外也可以先做两个尝试：
1）修改下evaluate.py，trainable改为False
2）把core/config.py 中TEST的SCORE 和 THRESHOLD阈值改小一点，看看能不能出结果，如果输出的框乱七八糟，说明你的模型不好，建议检查数据，重新多训练几个epoch，待loss收敛较小为止
感謝回復，有去降低threshold，是會預測出來且順利執行，但結果很不好，因為我的模型只用129張圖訓練，只想學會如何訓練，之後會去訓練多一點資料，想問可是我跑image_demo.py，單一張圖片時是可以有結果的，可是在evaluate.py中卻沒有結果呢，像是下圖是person_127的圖片，在image_demo.py下可以有結果，但在evaluate.py中卻是沒有結果。
我加了4處收縮，前兩行是指定GPU卡號，你主要改成pb_file（你要聲明的pb名，自己起一個），ckpt_file（選一個損失結果最好的）
想問如果用gpu訓練的話，是一定要打指定gpu那行嗎?
想問如果用gpu訓練的話，是一定要打指定gpu那行嗎?
不是必须指定GPU卡的，但如果不指定，默认会把所有的卡内存占满，别人想用其他卡的话就会出现内存溢出，如果你只有一个卡就没这个问题。如果多个卡中有一个是坏的，就会报错，我就遇到这种问题。另外，作者的代码是不支持多GPU的，就算打满多个卡，也只是用第一个卡在训练。
我換到實驗室的工作站跑，train完後轉成.pb檔，跑image_demo.py時出現下圖這個錯誤，想問是甚麼原因，因為這個問題想說是不是我問你的上個問題造成的，看起來好像不是
没有指定的话，会自动使用gpu训练吗，，能告诉我一下tensorflow-gpu版本吗，谢谢啦
-------- 原始邮件 -------- 发件人：cccgene0426 [email protected] 时间：2019年11月20日 17:06 收件人：YunYang1994/tensorflow-yolov3 [email protected] 抄送：MCQCQM [email protected],Comment [email protected] 主题：Re: [YunYang1994/tensorflow-yolov3] 自行用自己資料及訓練問題 (#400)
有關MOT的數據集可能要問作者，我訓練時沒有指定gpu
— You are receiving this because you commented. Reply to this email directly, view it on GitHub, or unsubscribe.
[ { "@context": "http://schema.org", "@type": "EmailMessage", "potentialAction": { "@type": "ViewAction", "target": "https://github.com/YunYang1994/tensorflow-yolov3/issues/400?email_source=notifications\u0026email_token=AKWRTAUKLJRAYB3RGENTDFLQUT423A5CNFSM4JMSVNS2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEERIFKY#issuecomment-555909803", "url": "https://github.com/YunYang1994/tensorflow-yolov3/issues/400?email_source=notifications\u0026email_token=AKWRTAUKLJRAYB3RGENTDFLQUT423A5CNFSM4JMSVNS2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEERIFKY#issuecomment-555909803", "name": "View Issue" }, "description": "View this Issue on GitHub", "publisher": { "@type": "Organization", "name": "GitHub", "url": "https://github.com" } } ]
跑freeze_graph.py檔案，改裡面的ckpt_file，改成你訓練完的.ckpt檔
scuizhibin [email protected] 於 2019年12月6日 週五 下午5:01寫道：
请问怎么生成PB文件呢？需要改动什么地方？
— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/YunYang1994/tensorflow-yolov3/issues/400?email_source=notifications&email_token=ANXXDOZJJECHGNSKYYPQWQTQXIIILA5CNFSM4JMSVNS2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEGDO3SA#issuecomment-562490824, or unsubscribe https://github.com/notifications/unsubscribe-auth/ANXXDO4OZTLRTUZNSPXXHELQXIIILANCNFSM4JMSVNSQ .
--
Testable And Reliable Systems Laboratory (TARS Lab)
Department of Electrical Engineering,
National Sun Yat-sen University
Kaohsiung, Taiwan
Tel: (07)5252000 Ext. 4123
Email: [email protected]
你好，我想问下你们用自己的数据训练了吗，我用自己的数据训练，按照博主的步骤走的，都跑通了，在最后测试的时候，我标记的是“人”这个类别，但是在运行evaluate.py出来的ground-truth跟predicted文件，里面出来的类别是“bird"最后运行main.py出来的也还是”bird"类别的ap，不知道哪里有问题。
@cccgene0426 你好，我训练自己数据集，在运行image_demo.py下有結果，但在evaluate.py中同样没有结果，请问下，你解决了吗？
還沒
wy-use [email protected] 於 2020年1月2日 週四 下午7:50寫道：
@cccgene0426 https://github.com/cccgene0426 你好，我训练自己数据集，在运行image_demo.py下有結果，但在evaluate.py中同样没有结果，请问下，你解决了吗？
— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/YunYang1994/tensorflow-yolov3/issues/400?email_source=notifications&email_token=ANXXDO4CEAI46F75JOFHXQ3Q3XIHXA5CNFSM4JMSVNS2YY3PNVWWK3TUL52HS4DFVREXG43VMVBW63LNMVXHJKTDN5WW2ZLOORPWSZGOEH6FUZY#issuecomment-570186343, or unsubscribe https://github.com/notifications/unsubscribe-auth/ANXXDO2JRZK45USEPXCDOGDQ3XIHXANCNFSM4JMSVNSQ .
--
Testable And Reliable Systems Laboratory (TARS Lab)
Department of Electrical Engineering,
National Sun Yat-sen University
Kaohsiung, Taiwan
Tel: (07)5252000 Ext. 4123
Email: [email protected]
我用電腦CPU跑時也是慢，但用GPU跑時就可以比較順
wanglei [email protected] 於 2020年3月18日 週三 上午11:48寫道：
我想请教一个问题，就是video_demo.py用视频流检测的时候非常慢，这个是和电脑硬件有关系吗?还是需要设置什么参数吗
— You are receiving this because you were mentioned. Reply to this email directly, view it on GitHub https://github.com/YunYang1994/tensorflow-yolov3/issues/400#issuecomment-600410526, or unsubscribe https://github.com/notifications/unsubscribe-auth/ANXXDO7RE5VPFJ4MWDFIZBDRIBAB7ANCNFSM4JMSVNSQ .
--
Testable And Reliable Systems Laboratory (TARS Lab)
Department of Electrical Engineering,
National Sun Yat-sen University
Kaohsiung, Taiwan
Tel: (07)5252000 Ext. 4123
Email: [email protected]
