- 博客(9)
- 资源 (5)
- 收藏
- 关注
RSS订阅原创 tensorflow统计算计和参数
参数数量：num_para=np.sum([np.prod(v.get_shape().as_list()) for v in tf.trainable_variables()])浮点运算量（mac或flops）：tf.contrib.tfprof.model_analyzer.print_model_analysis(tf.get_default_graph(), tfprof_options=tf.contrib.tfprof.model_analyzer.FLOAT_OPS_OPTIONS)或
2022-05-09 14:55:17
263
原创 用opencv函数替代skimage函数生成相似矩阵
人脸识别之前通常需要做校准工作，使得标准人脸和待识别人脸均位于"统一规范化"的条件下做比对；这里校准工作用的是相似变换SimilarityTransform【注：与仿射变换不同】，即：等距变换+均匀尺度缩放，而等距变换则是平移+旋转变换。对应的变换矩阵：目前大部分的做法是：采用python库中的skimage.transform模块生成变换矩阵，再利用opencv中的仿射变换函数cv2.warpAffine，将img生成"标准人脸"dst；过程如下：from skimage import trans
2021-04-19 19:26:10
1919
1
原创 windows下外接摄像头用python用于采集图像
笔记本外接摄像头，采集图像代码，import cv2import dlibfrom scipy.spatial import distancedef calculate_EAR(eye): A = distance.euclidean(eye[1], eye[5]) B = distance.euclidean(eye[2], eye[4]) C = distance.euclidean(eye[0], eye[3]) ear_aspect_ratio = (A+B)/(2.0*C) r
2021-02-07 14:51:01
629
原创 pytoch ModuleNotFoundError: No module named ‘models‘
在pytorch平台上，训练好模型。在推断时，推断文件路径发生了改变，结果报模型无法识别。net=torch.load('XXX.pt')错误如下return _load(f,map_location,pickle_module,**pickle_load_args)result = unpickler.load()ModuleNotFoundError: No module named 'models'根据网上资料：重新生成了’.tjm’可以解决我的问题，具体如下：def conve
2020-07-20 15:52:26
4166
6
原创 pytorch 载入模型参数finetune训练
pytorch 在finetune重训练时，采用torch.load()方式载入模型经常会报错。这里给出一种load方式，若模型中存在相同的tensor（名字和大小一致）则载入，否则对模型中tensor只做初始化处理。代码如下：def check_keys(model, pretrained_state_dict): ckpt_keys = set(pretrained_state_dict.keys()) model_keys= set(model.state_dict().key
2020-07-10 16:29:06
852
2
原创 pytorch平台图像数据处理
pytorch平台图像数据增强一般分两种：（1）调用opencv，采用cv2读入图像数据，自己定义数据增强函数，这种情况不考虑。（2）调用torchvision模块由于提供的torchvision主要使用PIL库中的Image类进行处理，所以它里面自带的图像数据增强函数，也均是以PIL读入的数据作为输入的。这时，如果要加入自己定义的函数，而且这些操作是cv2或者np.array情况下处理的画，需要转换格式。 i) PIL转numpy pil_img = Image.open(‘.jp
2020-07-10 15:49:21
508
原创 重载模型参数
tensorflow中做迁移学习时，会遇到模型参数值与网络模型参数定义不一致的问题，而无法载入模型参数。如参数名字改变，模型参数shape不一致，直接调用saver.restore()会报错。需要手动修改这个restore（）函数。这里提供两种恢复方式：（1）第一种方法，提取出ckpt中参数与模型中的参数名字一致，shape大小也一致，然后再调用saver.restore（）恢复。resto...
2019-12-31 11:21:30
156
原创 从tensorflow的pb文件中导出graph
从tensorflow保存的pb文件，解析出graph。代码：$ python -m tensorflow.python.tools.import_pb_to_tensorboard --model_dir=‘pb_path/model.pb’ --log_dir=‘logs_path’如：python -m tensorflow.python.tools.import_pb_to_ten...
2019-12-26 14:51:22
528
提取pascal voc数据中的一类或多类
2018-08-29
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人