資源簡介
本源碼是MDNet視頻目標跟蹤算法在Github上共享的、適用于Python 2.7版本的Python源碼。一種可行的運行環(huán)境為:CUDA 8.0.61 for Ubuntu 16.04、cuDNN 5.1 for CUDA 8.0、Python 2.7、PyTorch 0.2.0。
代碼片段和文件信息
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
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
????.......???????112??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\.gitignore
????.......?????????0??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\.gitkeep
????.......????????30??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\attrs.txt
????.......??????6798??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\cfg.json
????.......???????869??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\cfg.mat
????.......??????1490??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\groundtruth_rect.txt
????.......?????80538??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0001.jpg
????.......?????86649??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0002.jpg
????.......?????87579??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0003.jpg
????.......?????87743??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0004.jpg
????.......?????85623??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0005.jpg
????.......?????82421??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0006.jpg
????.......?????82730??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0007.jpg
????.......?????81100??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0008.jpg
????.......?????81830??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0009.jpg
????.......?????78536??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0010.jpg
????.......?????77630??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0011.jpg
????.......?????77141??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0012.jpg
????.......?????75724??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0013.jpg
????.......?????75934??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0014.jpg
????.......?????75277??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0015.jpg
????.......?????76822??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0016.jpg
????.......?????76316??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0017.jpg
????.......?????73082??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0018.jpg
????.......?????71938??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0019.jpg
????.......?????68982??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0020.jpg
????.......?????69480??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0021.jpg
????.......?????70347??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0022.jpg
????.......?????68931??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0023.jpg
????.......?????70270??2017-06-06?11:52??py-MDNet-master-for-Python2.7\py-MDNet-master\dataset\OTB\DragonBaby\img\0024.jpg
............此處省略120個文件信息
- 上一篇:python 3.7.3262322
- 下一篇:python mac包
評論
共有 條評論