哦哟这个怎么搞

cuda核简单了解

摘要： #include <stdio.h> #include <cuda_runtime.h> /* CUDA核函数 __global__ 核函数的前缀定义 - 使用__global__修饰的函数，必须是void无返回值 - __global__核函数修饰，必须是nvcc编译才有效，否则无效 - __gl 阅读全文

posted @ 2022-11-21 10:10 哦哟这个怎么搞阅读(197) 评论(0) 推荐(0)

YOLOV5 train模块写，没验证对错，能跑

摘要： import torch import torch.nn as nn import torch.optim as optim import dataset import models import nn_utils class YoloHead(nn.Module): def __init__(se 阅读全文

posted @ 2022-11-19 18:40 哦哟这个怎么搞阅读(31) 评论(0) 推荐(0)

cuda安装遇到you already have a new version of the nvidia frameview. have new

摘要： cuda11.2安装时和nvidia frameview sdk冲突怎么解决呀? - 知乎 (zhihu.com) 设置-应用-卸载nvidia frameview. 阅读全文

posted @ 2022-11-17 20:02 哦哟这个怎么搞阅读(4010) 评论(0) 推荐(0)

onnx可视化

摘要：工具网站 GitHub - lutzroeder/netron: Visualizer for neural network, deep learning, and machine learning models web在线网站 bottleneck.onnx (netron.app) 阅读全文

posted @ 2022-11-16 16:07 哦哟这个怎么搞阅读(167) 评论(0) 推荐(0)

yolov5dataset复写

摘要： import os import random import sys import cv2 import numpy as np from PIL import Image import sys_utils import nn_utils from sys_utils import _single_ 阅读全文

posted @ 2022-11-15 17:51 哦哟这个怎么搞阅读(29) 评论(0) 推荐(0)

annotation（N,5）映射到M矩阵的对应位置(2,3)

摘要：因为是pixel映射到对应的M矩阵对应位置，并且只需要前4个值（left,right,top,bottom） merge_mosaic_pixel_annotations = np.arange(0, 10).reshape(2, 5) print(f"merge_mosaic_pixel_anno 阅读全文

posted @ 2022-11-15 15:55 哦哟这个怎么搞阅读(62) 评论(0) 推荐(0)

yolov5中anchor的选择

摘要： import json import numpy as np import torch from scipy.cluster.vq import kmeans2,kmeans from sklearn.cluster import k_means import random with open("i 阅读全文

posted @ 2022-11-14 11:40 哦哟这个怎么搞阅读(119) 评论(0) 推荐(0)

简单的手写kmeans

摘要： k=2 data = np.random.randn(100,2) data[:50]+=2.0 center_index = np.random.choice(np.arange(len(data)), k) center = data[center_index] iters = 20 def d 阅读全文

posted @ 2022-11-09 10:09 哦哟这个怎么搞阅读(55) 评论(0) 推荐(0)

np小函数

摘要：按照面积排序 from scipy.cluster.vq import kmeans import numpy as np new_anchor = np.arange(0,9).reshape(3,3) print(new_anchor) print(new_anchor.prod(axis =1 阅读全文

posted @ 2022-11-07 08:57 哦哟这个怎么搞阅读(5) 评论(0) 推荐(0)

abl增广

摘要： from albumentations.core.transforms_interface import DualTransform from albumentations.pytorch import ToTensorV2, ToTensor import random View Code # 随阅读全文

posted @ 2022-11-03 08:07 哦哟这个怎么搞阅读(41) 评论(0) 推荐(0)

ruijiege

公告