目 录
XML文件操作(口罩检测)
CSV文件操作(口罩检测)
打印神经网络参数(EDSR-PyTorch)
net是网络的名称
def get_parameter_number(net):
total_num = sum(p.numel() for p in net.parameters())
trainable_num = sum(p.numel() for p in net.parameters() if p.requires_grad)
return {'Total': total_num, 'Trainable': trainable_num}
numbers = get_parameter_number(net)
print('Total parameters:',numbers['Total'],' ','Trainable parameters:',numbers['Trainable'])
经典网络及其模型的下载(蜜蜂蚂蚁pytorch官方迁移学习教程)
学习迁移学习的时候偶然发现的。针对的是torch框架。得到网络及模型。
pretrained = True时,下载网络及对应的模型,为False时,不下载预训练模型。
在指定位置下载(口罩检测)
import urllib.request
PRETRAINED_MODEL = './snapshots/resnet50_csv_02.h5'
URL_MODEL = 'https://github.com/fizyr/keras-retinanet/releases/download/0.5.1/resnet50_coco_best_v2.1.0.h5'
urllib.request.urlretrieve(URL_MODEL, PRETRAINED_MODEL)
import导入目录文件(SR)
Python-import导入上级目录文件
注意,这些操作都是相对于当前操作的文件。要以运行脚本的位置为中心来计算其他模块的导入。