随笔分类 -  Python

Yolov8 根据需要自定义颜色
摘要：1、找到系统文件 plotting.py，加一个类 Color class Colors: # Ultralytics color palette https://ultralytics.com/ def __init__(self): # hex = matplotlib.colors.TABLE 阅读全文
posted @ 2023-05-11 08:29 ~逍遥子~ 阅读(2256) 评论(1) 推荐(0) 编辑
Yolov5 根据自己的需要更改 预测框box和蒙版mask的颜色
摘要：1、首先找到项目中 utils/plots.py 文件，打开该代码 将原来的 Colors类注释掉（或删掉），改成如下Colors类 class Colors: def __init__(self): self.red = (255, 0, 0) # 深红色 self.green = (0, 255 阅读全文
posted @ 2023-05-10 20:10 ~逍遥子~ 阅读(1508) 评论(0) 推荐(0) 编辑
Python 利用 Pytorch 获取当前GPU信息
摘要：import torch def get_gpu_info(): device = torch.device("cuda" if torch.cuda.is_available() else "cpu") if device.type == "cuda": # 获取当前GPU名字 gpu_name 阅读全文
posted @ 2023-05-04 21:18 ~逍遥子~ 阅读(897) 评论(0) 推荐(0) 编辑
U2-Net 预测函数
摘要：包含单个图片检测以及视频检测 import os import time import cv2 import numpy as np import matplotlib.pyplot as plt import torch import time import subprocess from tor 阅读全文
posted @ 2023-04-28 13:17 ~逍遥子~ 阅读(33) 评论(0) 推荐(0) 编辑
Python通过终端 cmd 获取GPU相关信息
摘要：Linux终端实时观察 GPU 状态 每一秒输出一次，实时观察 watch -n 1 nvidia-smi Python代码获取 # 获取GPU相关信息 def get_gpu_info(): try: cmd_out = subprocess.check_output('nvidia-smi -- 阅读全文
posted @ 2023-04-28 13:16 ~逍遥子~ 阅读(86) 评论(0) 推荐(0) 编辑
Pandas 学习笔记
摘要：一、用Pandas创建Excel文件 import pandas as pd df = pd.DataFrame({'ID':[1,2,3],'Name':['liujun','daifen','duanziqian']}) # 创建一个数据集 df = df.set_index('ID') # 将 阅读全文
posted @ 2022-10-04 21:51 ~逍遥子~ 阅读(55) 评论(0) 推荐(0) 编辑