机器学习常用代码总结
爱吃肉der小兔叽
这个作者很懒,什么都没留下…
展开
-
cpu上训练的模型放到gpu上继续训练
cpu上训练的模型放到gpu上跑原创 2023-03-08 11:26:13 · 252 阅读 · 0 评论 -
mat1 dim 1 mustmatch mat2 dim0
极大可能是输入网络的图片维度有问题,一般在transform里面设置。原创 2023-01-30 13:15:08 · 186 阅读 · 0 评论 -
cutmix
import cv2 as cvimport matplotlib.pyplot as pltimport numpy as npimport randomimport numpy as npimport cv2import osimport mathimport torchfrom torch import nnfile_root = r’E:\wd_0.01_deal\wd_train_0.01\wd_train_0.01_4_2parts\real/’ # 当前文件夹下的所有图片原创 2022-11-02 21:46:00 · 189 阅读 · 0 评论 -
优化器的选择
优化器选择原创 2022-10-15 10:03:01 · 188 阅读 · 0 评论 -
遍历文件夹中的图片并进行处理并保存
py原创 2022-06-17 10:04:24 · 486 阅读 · 0 评论 -
pytorch保存模型与加载模型
模型保存与加载原创 2022-06-08 11:24:42 · 119 阅读 · 0 评论 -
resnet入手深度学习
import torchimport torch.nn as nnimport torch.optim as optimimport torchvisionimport torchvision.transforms as transformsimport argparsefrom resnet import resnet50# 定义是否使用GPUdevice = torch.device("cuda" if torch.cuda.is_available() else "cpu")#原创 2022-05-18 19:52:23 · 89 阅读 · 0 评论 -
python计算auc和rank
自己写的aucdef calAUC(prob,labels): f = list(zip(prob,labels)) rank = [values2 for values1,values2 in sorted(f,key=lambda x:x[0])] rankList = [i+1 for i in range(len(rank)) if rank[i]==1] posNum = 0 negNum = 0 for i in range(len(labels)): if(l原创 2022-05-06 10:12:05 · 211 阅读 · 0 评论 -
将文件夹中的子文件夹中的图片随机删除一半
import osimport randomrootdir = 'F://F++images//deepfakes_aligned_1'for root,dirs,files in os.walk(rootdir): for dir in dirs: print(os.path.join(root,dir)) num = 0; for file in files: feed = random.randint(0, 10) if原创 2022-03-11 16:12:50 · 424 阅读 · 0 评论 -
将图片从一个文件夹复制到另一个文件夹
import osimport randomimport shutil# source_file:源路径, target_ir:目标路径def cover_files(source_dir, target_ir): for file in os.listdir(source_dir): source_file = os.path.join(source_dir, file) if os.path.isfile(source_file):原创 2022-02-26 08:47:38 · 871 阅读 · 0 评论 -
python用代码划分数据集为训练集和测试集(从一个文件夹中随机剪切一定比例的图片到另一文件夹中)
import osimport randomimport shutil# source_file:源路径, target_ir:目标路径def cover_files(source_dir, target_ir): for file in os.listdir(source_dir): source_file = os.path.join(source_dir, file) if os.path.isfile(source_file):原创 2022-02-25 20:47:21 · 997 阅读 · 0 评论 -
制作真伪数据集脚本
#将文件夹中的图片归类到对应文件夹的子文件夹中,比如将大train中的train_real和train_fake中名称对应的图片转移到Image中的train的子文件夹对应的名称文件夹中(train Image test文件夹并列),具体的文件夹分布在微信收藏夹里面import osimport shutildef read_filename():file_path=‘F:/dealwith/train/train_fake’list=os.listdir(file_path)return li原创 2022-02-25 11:24:41 · 729 阅读 · 0 评论 -
将一个文件夹里面的视频批量转换为图片并保存在同一目录下
import osimport cv2cut_frame = 1 # 多少帧截一次,自己设置就行save_path = "/media/gpu/00758835-1a92-4611-8ae4-1518102a53d6/coding/pythonProject/pic"for root, dirs, files in os.walk(r"/media/gpu/00758835-1a92-4611-8ae4-1518102a53d6/coding/pythonProject/videos"):原创 2022-02-25 11:23:42 · 2943 阅读 · 1 评论