- 博客(18)
- 收藏
- 关注
原创 cuda10.1对应的tensorflow-gpu=2.2.0
电脑装了两个cuda版本,今天需要用到tensorflow-gpu=2.2.0,在环境变量里面已经把cuda10.1置顶了,但是怎么都调用不起来,在cmd里面输入nvcc -V显示的也是cuda10.1,但是就是用不了,只好把cuda装在conda环境里面了,输入下面这个就可以了conda install cudatoolkit=10.1 cudnn=7.6.5...
2022-04-26 20:23:27
1520
原创 画P-R曲线
import matplotlib.pyplot as pltimport numpy as npPrecisionbusoringin= [ 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00 , 1.00
2022-04-12 20:51:32
499
原创 batch-size较小时还能用batch normalization吗?
batch size比较小的时候还用batch normalization吗
2022-03-09 15:11:17
1228
原创 根据文件名称移动对应名称的图片
比如我有一堆txt后缀的文件,如下图我需要找到这些txt对应的图片,使用如下代码import osimages_dir = r'VOCdevkit/VOC2007/JPEGImages'txt_dir = r'D:\GoogleDownload\python_project_file\yolov4-tiny-keras-voc\map_out\ground-truth'import shutil# 创建列表txts = []# 读取xml文件名(即:标注的图片名)for txt in
2022-03-02 13:00:40
347
1
原创 YOLOx添加Focal loss
将YOLOv3及以上的网络中的BCE loss更改为Focal lossloss函数分为三部分,位置损失、置信度损失、类别损失,此处只需要将置信度损失更换为Focal loss,具体原理请仔细理解置信度损失的含义。YOLOX链接:https://link.zhihu.com/?target=https%3A//github.com/Megvii-BaseDetection/YOLOX1 找到置信度预测损失计算位置loss_obj,并进行替换(位置在386-405行左右)loss_iou:定位损失;l
2022-02-25 20:20:41
6238
12
原创 对一个文件夹下的所有图片进行数据增强
一张图片可以扩充七张# coding = utf-8import cv2from PIL import Imagefrom PIL import ImageEnhancefrom numpy.ma import arrayimport numpy as npimport os# 批量处理代码rootdir = 'D:/GoogleDownload/GrapeDeseaseDataset/Grape_Black_rot' # 指明被遍历的文件夹def high_bright(curren
2022-01-05 12:21:26
2443
4
转载 yolov3 h5转pb
修改自己的类个数即可https://github.com/nistarlwc/keras-yolo3-convert2pb
2021-06-23 10:37:15
180
1
原创 查看tensorflow的pb文件里每一层的结构名
这里写自定义目录标题欢迎使用Markdown编辑器新的改变功能快捷键合理的创建标题,有助于目录的生成如何改变文本的样式插入链接与图片如何插入一段漂亮的代码片生成一个适合你的列表创建一个表格设定内容居中、居左、居右SmartyPants创建一个自定义列表如何创建一个注脚注释也是必不可少的KaTeX数学公式新的甘特图功能,丰富你的文章UML 图表FLowchart流程图导出与导入导出导入欢迎使用Markdown编辑器import tensorflow as tfimport osmodel_dir =
2021-06-23 10:34:16
296
原创 python去除文件夹下文件名中的空格
把path路径的内容换成自己的就可以了import osprint("**********************************************")path = " ";# walk方法会返回一个三元组,分别是root、dirs和files。# 其中root是当前正在遍历的目录路径;dirs是一个列表,包含当前正在遍历的目录下所有# 的子目录名称,不包含该目录下的文件;files也是一个列表,包含当前正在遍历的# 目录下所有的文件,for root, dirs, file
2021-05-17 21:17:55
1195
原创 yolov3 .h5转.pb文件(keras)
yolov3转.pb文件(keras)我是先转的weight,再转的.pb。转weights参考https://github.com/amir-abdi/keras_to_tensorflow.git和博客https://blog.csdn.net/python_pycharm/article/details/106804233,转.pb参考https://github.com/YunYang1994/tensorflow-yolov3.git和博客https://blog.csdn.net/weixin
2021-05-17 21:15:28
462
原创 查看keras模型的输入输出节点名称
查看keras模型的输入输出节点名称有两种方式1.直接调用model.inputs和model.outputs2.利用.pb文件查看,代码如下import tensorflow as tfimport osmodel_dir = './'model_name = 'frozen_graph.pb'def create_graph(): with tf.gfile.FastGFile(os.path.join( model_dir, model_nam
2021-05-17 21:07:33
999
2
手工标注葡萄叶片病害数据集
2022-09-30
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人