![](https://img-blog.csdnimg.cn/20201014180756738.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch学习
文章平均质量分 60
lth在海上漂
unity开发工作者,深度学习开发工作者
展开
-
pytorch-上分之路-图像分割
目录项目结构commonconfigdatalistmodelutilstrain总结项目结构这个项目的数据集主要来自百度的道路分割,官网可以下到。具体代码内容如下commonimport torch.nn as nnclass ConvBNReLU(nn.Sequential): def __init__(self, inplanes, planes, kernel_size, padding, dilation): super(ConvBNReLU, self).原创 2021-03-27 19:51:38 · 504 阅读 · 1 评论 -
pytorch上分之路-yolov4
目录项目结构configdatalistmodelutilstrain总结项目结构这个是我做yolo项目的结构,总结一下,其他都中规中矩,比较麻烦的是数据的转换和loss的计算,不像我之前的项目,感觉自己一个人写不出来,主要感觉是太繁琐了,所以找了别人的github项目中的部分内容借鉴了一下特别注意,你在使用这个项目之前要把数据集,class和anchor的txt准备好。configfrom model import YoloBodyfrom config import parserimp原创 2021-03-27 19:40:54 · 210 阅读 · 0 评论 -
LPR车牌识别-pytorch上分之路
LPRNet-车牌识别项目目录config.pydatalist.pymodel.pyutils.pytrain.pytrainDDP.py最后项目目录这里主要就看我在目录里强调的就行了,其他的是我练手瞎写的config.pyimport argparsedef get_parser(): parser = argparse.ArgumentParser(description='parameters to train net') parser.add_argument('--原创 2021-02-22 22:22:53 · 1961 阅读 · 10 评论 -
3dcnn视频分类算法-pytorch上分之路
3DCNN-视频分类项目结构config.pydatalist.pymodel.pytrain.py最后项目结构config.pyimport argparse'''training settingsmetavar参数,用来控制部分命令行参数的显示'''parser = argparse.ArgumentParser(description='PyTorch Example for all')parser.add_argument('--train-batch-size', type原创 2021-02-22 21:59:52 · 4748 阅读 · 14 评论 -
pytorch上分之路——视频补全算法(onion peel network)
文章目录前言一、config.py二、datalist.py三.common.py四.model.py五.model_common.py六.train.py总结前言该算法是从github上找的onion peel network算法,但是由于开发者只提供了demo部分,所以我试着自己把train的部分自己实现了,目前来看多少有点能补全的意思。目前来看还不是很成熟,但我还是发出来给大家看看。当然我把风格严格控制成我以前发的代码风格,方便学习和以后的优化一、config.pyimport argpar原创 2020-12-25 21:31:55 · 981 阅读 · 4 评论 -
PyTorch-上分之路(基于Vision Transformer的鸟群分类算法)
目录前言一、config.py二、datalist.py1.引入库三.model.py四、utils.py五、train.py六、总结前言之前发了一个shufflenet的鸟群分类算法,以fintune的方法来做,效果只有56%,太差了,所以我换了一个模型,又加入了一些微调手段来提神效果,现在把代码分享出来,供各位看官指正,希望能让我的分类结果进一步提升从这里可以看出,我使用的vision transformer模型是最小的,因为能用的只有一个2080分享的代码里模型部分是参考了github的原创 2020-12-13 14:50:35 · 2810 阅读 · 10 评论 -
Pytorch上分之路—ShuffleNetv3(鸟群分类算法)
Pytorch上分之路—ShuffleNetv3(鸟群分类算法)本次的内容是用pytorch写一个简单的分类算法,选择了200鸟群的数据集,数据集的话可以自己到网上去找,挺容易的。目录**Pytorch上分之路—ShuffleNetv3(鸟群分类算法)**项目结构一、config二、datalist三.ShuffleNet四 train五 utils六 inference项目结构项目中所有的文件组成config.py用于配置一些主要的参数datalist.py用于数据的输出infer原创 2020-12-01 22:24:24 · 2717 阅读 · 0 评论 -
Torch上分之路——Mnist
Torch上分之路——Mnist文档目录(其中common用于写模型,utils用于写算法过程中要用到函数工具,所以只有datalist,model,train,config会被用到config.py模型配置文件import argparse# Training settings#metavar参数,用来控制部分命令行参数的显示parser = argparse.ArgumentParser(description='PyTorch MNIST Example')parser.add_a原创 2020-11-07 16:32:53 · 842 阅读 · 0 评论