- 博客(9)
- 收藏
- 关注
RSS订阅原创 图像分割的one-hot
以这个标签（class=2）即左上角第一个数据（红色框）为例，由于dim=1，所以将src=1赋值给one-hot中索引为 [0, 2, 0, 0] 这个位置 （前面第一个位置 0是 batch_size索引，由于这里选择第一个batch，所以索引是0，第二个位置2是标签索引，也就是class=2，后面的h和w是原来标签所在的h和w位置）。同理，绿色框的数据和蓝色框的数据）。这里只对四维数据分析one-hot，即 (batch_size, num_class, h, w) ------图像------
2024-09-19 20:34:07
243
原创 torch笔记-----torch.nn.embedding的理解
可以理解为对于输入x，有4个句子，其中每个句子有10个单词，然后通过nn.embedding操作，将每个词转换成一个个向量，而这些向量是从nn.embedding中查询（每个单词会有一个索引，然后通过索引进行查询）得到的，其中nn.embedding存储的有200个单词，每个单词的向量长度是20，所以最后输出的y.shape(batch_size=4, seq_len=10, embedding_dim=20)。例如：输入x.shape(batch_size=4，seq_len=10)
2024-09-02 18:49:11
543
原创 Mamba YOLO环境配置和测试（Ubuntu20.04-cuda11.8）
下载了selective-scan.whl (CUDA 11.8 包含selective_scan_core），并安装。注：如果不安装 selective_scan_core，也可以进行测试，但会出现这个错误！，也可以按照官方Readme要求进行配置，但。
2024-08-23 21:19:28
396
原创 每日论文阅读01
MambaHSI: Spatial–Spectral Mamba for Hyperspectral Image Classification论文阅读
2024-08-13 21:36:30
601
1TA创建的收藏夹 TA关注的收藏夹
TA关注的人