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-master\LICENSE
?????文件????????1466??2017-06-06?03:52??py-MDNet-master\README.md
?????目錄???????????0??2017-06-06?03:52??py-MDNet-master\dataset\
?????目錄???????????0??2017-06-06?03:52??py-MDNet-master\dataset\OTB\
?????目錄???????????0??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\
?????文件???????????0??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\.gitkeep
?????文件??????????30??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\attrs.txt
?????文件????????6798??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\cfg.json
?????文件?????????869??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\cfg.mat
?????文件????????1490??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\groundtruth_rect.txt
?????目錄???????????0??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\
?????文件???????80538??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0001.jpg
?????文件???????86649??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0002.jpg
?????文件???????87579??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0003.jpg
?????文件???????87743??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0004.jpg
?????文件???????85623??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0005.jpg
?????文件???????82421??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0006.jpg
?????文件???????82730??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0007.jpg
?????文件???????81100??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0008.jpg
?????文件???????81830??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0009.jpg
?????文件???????78536??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0010.jpg
?????文件???????77630??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0011.jpg
?????文件???????77141??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0012.jpg
?????文件???????75724??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0013.jpg
?????文件???????75934??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0014.jpg
?????文件???????75277??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0015.jpg
?????文件???????76822??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0016.jpg
?????文件???????76316??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0017.jpg
?????文件???????73082??2017-06-06?03:52??py-MDNet-master\dataset\OTB\DragonBaby\img\0018.jpg
............此處省略115個文件信息