摘要:
日志代码没有解释,已在train_classification中,且该代码是在GPU下运行 点击查看代码 """ Author: Benny Date: Nov 2019 """ import argparse import os import torch import datetime impor 阅读全文
posted @ 2022-01-13 15:17
原来是只呆燕
阅读(179)
评论(0)
推荐(0)
摘要:
1、1*1的卷积核有啥作用 答:主要是降维或升维,也就是多个feature channels线性叠加,比如,一张256×256且深度depth为50的图 (256×256×50)在20个filter上做1×1的卷积,结果会得到256×256×20的图片(降维),若在80个filter上做1×1的卷积 阅读全文
posted @ 2022-01-13 10:35
原来是只呆燕
阅读(38)
评论(0)
推荐(0)
摘要:
1、1-hot encodes a tensor 点击查看代码 def to_categorical(y, num_classes): """ 1-hot encodes a tensor """ new_y = torch.eye(num_classes)[y.cpu().data.numpy() 阅读全文
posted @ 2022-01-10 19:30
原来是只呆燕
阅读(54)
评论(0)
推荐(0)
摘要:
下面的代码与之前的代码有很多重复之处,就不在一一注释,只注释个人还不太熟悉的 点击查看代码 import torch.nn as nn import torch.utils.data import torch.nn.functional as F from models.pointnet_utils 阅读全文
posted @ 2022-01-09 16:58
原来是只呆燕
阅读(59)
评论(0)
推荐(0)
摘要:
点击查看代码 # *_*coding:utf-8 *_* import os import json import warnings import numpy as np from torch.utils.data import Dataset warnings.filterwarnings('ig 阅读全文
posted @ 2022-01-06 16:39
原来是只呆燕
阅读(233)
评论(0)
推荐(0)
摘要:
点击查看代码 #下面的代码中的一些超参数并没有按照原论文中来,不过总体结构差不多 """ Author: Benny Date: Nov 2019 """ import os import sys import torch import numpy as np import datetime #获取 阅读全文
posted @ 2022-01-03 09:37
原来是只呆燕
阅读(211)
评论(0)
推荐(0)