YOLO-V3-IOU issues

使用gpu训练loss一直是nan

使用gpu训练loss一直是nan，cpu则可以正常运行，怀疑是版本问题，请问作者用的什么版本的keras和tensorflow

重新训练问题

1

请问设置迭代20次，每次到第6次或第7次就会强制退出是怎么回事，训练样本42500，学习率设置0.001，其他的都一样。

zhangxuexin1995

您好，我将程序clone到了本地，没有用训练好的weight.h5和model.h5，将load_pretrained改为false之后，学习率改为0.001，迭代次数为50次，最后得到的权重来重新运行run.py，但结果确实绝大多数图片没有检测框，出现的几个位置也很明显不对。我看了我最后的loss和val_loss，均为3左右。请问为什么会出现这种结果呢，数据集和class、anchor这些均没有变过。同时我用您训练好的确实是有很好的检测结果的，所以想请教具体的训练参数是什么呢。

ppx-hub

关于拿自己的数据进行训练却出不了预测框

7

您好，代码我顺利运行了，检测动漫人脸的效果很好，然后想把它用在检测浣熊上，觉得都是单目标检测，应该没多少需要改变的我采用了这个数据集：https://github.com/datitran/raccoon_dataset 然后把图像和标签分别放到images和annotations下，运行了kmeans.py生成相应的anchors，存入infos/anchors.txt，将infos/classes.txt的内容改成raccoon 运行prepare.py将数据集进行划分，在infos文件夹下生成几个txt文件其他基本就没有改动了直接运行train.py，50轮迭代后loss降到18，应该属于还行的程度了但调用yolo.detect_image()后却出不了框，所有的测试集都没有框，没有报错内容基本的文件组织方式都是模仿这个项目的，哪个文件夹下放哪些文件都是模仿的，数据集复杂程度也差不多，都是单目标检测，但最后出不了效果，请问我在模仿时是漏关注什么地方了吗？

yan4821567

GPU

这个代码怎么限制GPU使用情况，我用 config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.1 session = tf.Session(config=config) self.sess = K.get_session(session) 这样是错误的，作者大神有什么方法限制显存么

zhonganzuzhihu