MDNETpython实现
MDNET论文的pytorch实现代码,python版本python2.7,相关论文请结合我的博客进行阅读。
代码片段和文件信息
属性 大小 日期 时间 名称
----------- --------- ---------- ----- ----
目录 0 2017-06-06 03:52 py-MDNet-master
文件 112 2017-06-06 03:52 py-MDNet-master.gitignore
文件 2597 2017-06-06 03:52 py-MDNet-masterLICENSE
文件 1466 2017-06-06 03:52 py-MDNet-masterREADME.md
目录 0 2017-06-06 03:52 py-MDNet-masterdataset
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTB
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBaby
文件 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBaby.gitkeep
文件 30 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyattrs.txt
文件 6798 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabycfg.json
文件 869 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabycfg.mat
文件 1490 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabygroundtruth_rect.txt
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg
文件 80538 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 001.jpg
文件 86649 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 002.jpg
文件 87579 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 003.jpg
文件 87743 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 004.jpg
文件 85623 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 005.jpg
文件 82421 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 006.jpg
文件 82730 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 007.jpg
文件 81100 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 008.jpg
文件 81830 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 009.jpg
文件 78536 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 010.jpg
文件 77630 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 011.jpg
文件 77141 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 012.jpg
文件 75724 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 013.jpg
文件 75934 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 014.jpg
文件 75277 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 015.jpg
文件 76822 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 016.jpg
文件 76316 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 017.jpg
文件 73082 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 018.jpg
............此处省略115个文件信息
import os
import scipy.io
import numpy as np
from collections import OrderedDict
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
import torch
def append_params(params module prefix):
for child in module.children():
for kp in child._parameters.iteritems():
if p is None: continue
if isinstance(child nn.BatchNorm2d):
name = prefix + ‘_bn_‘ + k
else:
name = prefix + ‘_‘ + k
if name not in params:
params[name] = p
else:
raise RuntimeError(“Duplicated param name: %s“ % (name))
class LRN(nn.Module):
def __init__(self):
super(LRN self).__init__()
def forward(self x):
#
# x: N x C x H x W
pad = Variable(x.data.new(x.size(0) 1 1 x.size(2) x.size(3)).zero_())
x_sq = (x**2).unsqueeze(dim=1)
x_tile = torch.cat((torch.cat((x_sqpadpadpadpad)2)
torch.cat((padx_sqpadpadpad)2)
torch.cat((padpadx_sqpadpad)2)
torch.cat((padpadpadx_sqpad)2)
torch.cat((padpadpadpadx_sq)2))1)
x_sumsq = x_tile.sum(dim=1).squeeze(dim=1)[:2:-2::]
x = x / ((2.+0.0001*x_sumsq)**0.75)
return x
class MDNet(nn.Module):
def __init__(self model_path=None K=1):
super(MDNet self).__init__()
self.K = K
self.layers = nn.Sequential(OrderedDict([
(‘conv1‘ nn.Sequential(nn.Conv2d(3 96 kernel_size=7 stride=2)
nn.ReLU()
LRN()
nn.MaxPool2d(kernel_size=3 stride=2)))
(‘conv2‘ nn.Sequential(nn.Conv2d(96 256 kernel_size=5 stride=2)
nn.ReLU()
LRN()
nn.MaxPool2d(kernel_size=3 stride=2)))
(‘conv3‘ nn.Sequential(nn.Conv2d(256 512 kernel_size=3 stride=1)
nn.ReLU()))
(‘fc4‘ nn.Sequential(nn.Dropout(0.5)
nn.Linear(512 * 3 * 3 512)
nn.ReLU()))
(‘fc5‘ nn.Sequential(nn.Dropout(0.5)
nn.Linear(512 512)
nn.ReLU()))]))
self.branches = nn.ModuleList([nn.Sequential(nn.Dropout(0.5)
nn.Linear(512 2)) for _ in range(K)])
if model_path is not None:
if os.path.splitext(model_path)[1] == ‘.pth‘:
self.load_model(model_path)
elif os.path.splitext(model_path)[1] == ‘.mat‘:
sel
属性 大小 日期 时间 名称
----------- --------- ---------- ----- ----
目录 0 2017-06-06 03:52 py-MDNet-master
文件 112 2017-06-06 03:52 py-MDNet-master.gitignore
文件 2597 2017-06-06 03:52 py-MDNet-masterLICENSE
文件 1466 2017-06-06 03:52 py-MDNet-masterREADME.md
目录 0 2017-06-06 03:52 py-MDNet-masterdataset
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTB
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBaby
文件 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBaby.gitkeep
文件 30 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyattrs.txt
文件 6798 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabycfg.json
文件 869 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabycfg.mat
文件 1490 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabygroundtruth_rect.txt
目录 0 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg
文件 80538 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 001.jpg
文件 86649 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 002.jpg
文件 87579 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 003.jpg
文件 87743 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 004.jpg
文件 85623 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 005.jpg
文件 82421 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 006.jpg
文件 82730 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 007.jpg
文件 81100 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 008.jpg
文件 81830 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 009.jpg
文件 78536 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 010.jpg
文件 77630 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 011.jpg
文件 77141 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 012.jpg
文件 75724 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 013.jpg
文件 75934 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 014.jpg
文件 75277 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 015.jpg
文件 76822 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 016.jpg
文件 76316 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 017.jpg
文件 73082 2017-06-06 03:52 py-MDNet-masterdatasetOTBDragonBabyimg 018.jpg
............此处省略115个文件信息
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件举报,一经查实,本站将立刻删除。
评论列表(条)