tensorflow-yolov3
tensorflow-yolov3 copied to clipboard
🔥 TensorFlow Code for technical report: "YOLOv3: An Incremental Improvement"
Hi YunYang, Good job! How to added background image to train? Is it added the background image path the the train.txt and not providing the box and class information? Thanks
当我把video_demo.py中权重加载路径改成自己训练的权重文件之后进行预测的时总是会报错如下: tensorflow.python.framework.errors_impl.NotFoundError: /home/GanYB/PycharmProjects/object_detection/tensorflow-yolov3/checkpoint/yolov3_train_loss=6.1510.ckpt-50; No such file or directory 请问这个问题怎么解决呢
2021-01-20 17:01:33.778610: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine. Traceback (most recent call last): File "D:/python-workspace/tensorflow-yolov3-master/convert_weight.py", line 30, in...
hi @YunYang1994 谢谢分享代码,我想请问下,我的图像尺寸是1000多*1000多的。我用docs/box_clustering对归一化后的宽和高聚类了。聚类中心都是0-1的值,我需要按照图像尺寸还原到真实大小,再把这些值当成我的数据的anchor吗? 还是直接用您提供的baseline的anchor或者coco的anchor就可以? 谢谢
ckpt训练好后,跑evaluate.py结果正常,速度也正常。 转成pb文件后,再用image_demo验证,会出来几百个结果,其中包含一个正确的结果,其他都是不合理的值,比如得分超过1,box范围超出界限等等。 在后面加一段滤除不合理结果的代码可以解决效果问题,但可能也是结果太多的原因,在ckpt下只需要几十毫秒就能跑完一张图,在pb下需要几秒。 怀疑是ckpt frozen成pb时出了问题,有人碰到这问题吗?多谢解答
自行用自己資料及訓練問題
自行用自己的資料庫訓練,訓練完有產生yolov3_test_loss=17.2442.ckpt-30這種meta檔案,之後還要跑convert_weight.py和freeze_weight.py這兩個檔案,但在跑convert_weight.py時出現runtimeerror的錯誤,想問在檔案裡有甚麼地方需要修改的,謝謝。 
dataset.txt
I do not understand how to make dataset.txt? Could you please tell me about this? thanks.
我用1660s跑训练,batchsize最大是2,再增加就会显存溢出,有什么办法跑大batch吗,难道6G显存只能跑batchsize=2么,感谢
你好,在dataset.py的第217行‘anchors_xywh[:, 0:2] = np.floor(bbox_xywh_scaled[i, 0:2]).astype(np.int32) + 0.5’,看代码应该是给anchors设置和bbox一样的center再计算iou,这里为什么不直接用bbox_xywh_scaled的中心,而是先向下取整,再转为int32,最后再加上0.5?我猜是和python的浮点数计算有关,但是不确定
- OpenCV => 4.4.0 - Operating System / Platform => macOS - Compiler => Visual Studio Version: 1.52.1 ##### Detailed description I am training yolo model on my set of...