
pytorch
国产零零零七
这个作者很懒,什么都没留下…
展开
-
pytorch merge bn (model from mmdetec)
import osimport torchsrc_weight = 'epoch_12.pth'model = torch.load(src_weight)weight = model['state_dict']filter_names = []layer_names = list(weight.keys())for layer_name in layer_names: ...原创 2019-10-15 01:18:27 · 381 阅读 · 0 评论 -
Win10/Ubuntu16.04+RTX2070+Cuda10+deepin-wine软件+Anaconda+Pycharm2018+Pytorch1.0+faster-rcnn
前言记笔记真的非常有必要,因为好记性不如烂笔头,曾经填过的坑不仅可以帮助爬坑人,也可以帮助自己日后不小心又掉进坑的时候,能够快速爬出来!!!目录 Ubuntu16.04LTS安装+磁盘分区 NVIDIA显卡驱动安装+CUDA10.0/CUDNN7.4安装 deepin-wine软件安装 Anaconda3 安装 Pycharm2018专业版安装...原创 2019-02-27 12:10:01 · 2994 阅读 · 3 评论 -
simple-faster-rcnn-pytorch 训练自己的数据 (pascal voc格式)
项目地址:simple-faster-rcnn-pytorch修改脚本:将voc_dataset.py和vis_tool.py里面的VOC_BBOX_LABEL_NAMES类别替换成自己标签中类别名称,必须和xml文件里面的标签名字一致,否则会直接跳过这个样本。 修改config.py中的config.py中的voc_data_dir路径,改成自己的路径. 修改faster_rcnn_...原创 2019-04-02 14:41:59 · 2411 阅读 · 8 评论