資源簡介
這是基于The Unreasonable Effectiveness of Deep Features as a Perceptual Metric 中所提到的感知相似度計算網絡代碼,里面有鏈接可供下載作者訓練好的權重。供大家參考

代碼片段和文件信息
import?argparse
from?models?import?dist_model?as?dm
from?util?import?util
parser?=?argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
parser.add_argument(‘--path0‘?type=str?default=‘./imgs/ex_ref.png‘)
parser.add_argument(‘--path1‘?type=str?default=‘./imgs/ex_p0.png‘)
parser.add_argument(‘--use_gpu‘?action=‘store_true‘?help=‘turn?on?flag?to?use?GPU‘)
opt?=?parser.parse_args()
##?Initializing?the?model
model?=?dm.DistModel()
model.initialize(model=‘net-lin‘net=‘alex‘use_gpu=opt.use_gpu)
#?Load?images
img0?=?util.im2tensor(util.load_image(opt.path0))?#?RGB?image?from?[-11]
img1?=?util.im2tensor(util.load_image(opt.path1))
#?Compute?distance
dist01?=?model.forward(img0img1)
print(‘Distance:?%.3f‘%dist01)
?屬性????????????大小?????日期????時間???名稱
-----------?---------??----------?-----??----
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\
?????文件??????????21??2018-11-25?05:46??PerceptualSimilarity-master\.gitignore
?????文件????????1671??2018-11-25?05:46??PerceptualSimilarity-master\Dockerfile
?????文件????????1357??2018-11-25?05:46??PerceptualSimilarity-master\LICENSE
?????文件????????9216??2018-11-25?05:46??PerceptualSimilarity-master\README.md
?????文件????????1142??2018-11-25?05:46??PerceptualSimilarity-master\__init__.py
?????文件?????????754??2018-11-25?05:46??PerceptualSimilarity-master\compute_dists.py
?????文件????????1097??2018-11-25?05:46??PerceptualSimilarity-master\compute_dists_dirs.py
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\data\
?????文件???????????0??2018-11-25?05:46??PerceptualSimilarity-master\data\__init__.py
?????文件?????????167??2018-11-25?05:46??PerceptualSimilarity-master\data\ba
?????文件????????1482??2018-11-25?05:46??PerceptualSimilarity-master\data\custom_dataset_data_loader.py
?????文件?????????476??2018-11-25?05:46??PerceptualSimilarity-master\data\data_loader.py
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\data\dataset\
?????文件???????????0??2018-11-25?05:46??PerceptualSimilarity-master\data\dataset\__init__.py
?????文件?????????237??2018-11-25?05:46??PerceptualSimilarity-master\data\dataset\ba
?????文件????????1792??2018-11-25?05:46??PerceptualSimilarity-master\data\dataset\jnd_dataset.py
?????文件????????2411??2018-11-25?05:46??PerceptualSimilarity-master\data\dataset\twoafc_dataset.py
?????文件????????2261??2018-11-25?05:46??PerceptualSimilarity-master\data\image_folder.py
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\imgs\
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir0\
?????文件????????9464??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir0\0.png
?????文件????????9464??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir0\1.png
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir1\
?????文件????????7518??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir1\0.png
?????文件???????10078??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_dir1\1.png
?????文件????????7518??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_p0.png
?????文件???????10078??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_p1.png
?????文件????????9464??2018-11-25?05:46??PerceptualSimilarity-master\imgs\ex_ref.png
?????文件??????843458??2018-11-25?05:46??PerceptualSimilarity-master\imgs\fig1.png
?????目錄???????????0??2018-11-25?05:46??PerceptualSimilarity-master\models\
............此處省略35個文件信息
- 上一篇:Pytorch模型權重轉變為Keras對應的模型權重
- 下一篇:霍爾測速和顯示
評論
共有 條評論