ZJ-项目中深度学习小知识汇总

XML文件操作（口罩检测）

python 读取与修改 XML（增删改查）

CSV文件操作（口罩检测）

使用pandas创建、修改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导入上级目录文件
注意，这些操作都是相对于当前操作的文件。要以运行脚本的位置为中心来计算其他模块的导入。