深度学习
SeanQikYibo
这个作者很懒,什么都没留下…
展开
-
深度学习预训练模型下载
'alexnet': 'https://download.pytorch.org/models/alexnet-owt-4df8aa71.pth', 'densenet121': 'http://data.lip6.fr/cadene/pretrainedmodels/densenet121-fbdb23505.pth', 'densenet169': 'http://data.lip6.fr/cadene/pretrainedmodels/densenet169-f470b90a...原创 2021-05-18 09:23:38 · 720 阅读 · 1 评论 -
深度学习分类数据集小工具
1、获取分类trian目录下各类别的图像数量import osdir = r"./data/train"dirs = os.listdir(dir)for i in range(len(dirs)): try: imgs = os.listdir(os.path.join(dir, dirs[i])) print(dirs[i]+":"+str(len(imgs))) except: pass2、将train数据集按照比例随原创 2021-01-15 14:09:14 · 232 阅读 · 0 评论 -
ESRGAN训练数据
一、ESRGAN测试:1、代码下载:xinntao / ESRGAN2、安装环境(这里略)3、下载models:Google DriveorBaidu Drive. 然后把它们放在./models文件夹下面。4、 然后运行test.py,就会将LR文件夹的低分辨率图像生成X4(可更改model)的高分辨率图像存在results文件夹内二、ESRGAN训练1、代码下载xinntao / BasicSR2、下载预训练模型,下载数据集DIV2K,或者其他常见图像超分数据集3..原创 2021-01-14 21:48:15 · 6928 阅读 · 24 评论 -
EfficientNet 训练自己的分类数据集
参考乱觉先森的这篇文章:EfficientNet 训练测试自己的分类数据集1、下载代码Levigty/EfficientNet-Pytorch可快速使用。(这里顺便提一下EfficientNet的pytorch版官方代码)2、准备数据集格式如下:3、下载预训练模型可复制链接至迅雷下载更快https://github.com/lukemelas/EfficientNet-PyTorch/releases/download/1.0/efficientnet-b0-355c32eb..原创 2021-01-12 15:45:58 · 3721 阅读 · 9 评论 -
deep_sort_yolov3实现多人目标跟踪
1、下载keras-yolov3:https://github.com/qqwweee/keras-yolo3下载yolov3.weights:https://pjreddie.com/media/files/yolov3.weights将其放在keras-yolo3目录下然后在keras-yolo3文件目录运行一下命令,得到yolo.h5python convert.py yolov3.cfg yolov3.weights model_data/yolo.h5(中间要安装Numpy、Te原创 2020-12-12 21:48:40 · 392 阅读 · 0 评论