深度学习
深度学习
curtain灬、
这个作者很懒,什么都没留下…
展开
-
yolov3物体检测
文件组织架构checkpoints 训练后的模型config 配置文件data 数据集logs 日志output 输出预测(运行detect.py后生成的图片)utils 使用的函数 weights 权重模型models.py #模型train.py 训练test.py 测试detect.py 快速使用模型requirements.txt ...原创 2020-02-06 14:24:58 · 2066 阅读 · 1 评论 -
pytorch物体检测(视频检测)
今天参考github上的一个项目完成了视频检测https://github.com/ayooshkathuria/pytorch-yolo-v3准备工作首先把相关文件复制到自己项目的根目录下还需下载pandas,这里直接用匹配命令pip install pandas -i http://pypi.douban.com/simple --trusted-host pypi.douban.c...原创 2020-02-07 15:46:52 · 3528 阅读 · 1 评论 -
Pytorch 实现MNIST
MNIST 包括6万张28x28的训练样本,1万张测试样本,本文使用的CNN网络导入包import torchimport torch.nn as nnimport torch.nn.functional as Fimport torch.optim as optimfrom torchvision import datasets, transformsfrom torch.auto...原创 2020-01-31 15:03:35 · 245 阅读 · 0 评论 -
pytorch模型提示超出内存cuda runtime error(2): out of memory
网上查阅了一些资料,包括什么改虚拟内存,都没什么用,最后我修改了batch_size,把它改小,模型终于能跑了原创 2020-02-03 15:44:04 · 1213 阅读 · 0 评论 -
UserWarning: indexing with dtype torch.uint8 is now deprecated, please use a dtype torch.bool instea
今天用pytorch训练的时候,出现了一堆这样的报错,翻阅了一些资料,找到了解决方法找到model.py将obj_mask改为obj_mask.bool()将noobj_mask改为noobj——mask.bool()我在repalce all的时候,发现部分已经加了.bool()所以在替换的时候要注意一些...原创 2020-02-02 18:15:38 · 4527 阅读 · 1 评论 -
RuntimeError: cannot perform reduction function max on tensor with no elements because the operation
今天运行train遇到这个问题,后面检查发现,我的label中有空的txt文件,重新生成label后解决原创 2020-02-04 15:25:34 · 9924 阅读 · 6 评论