- 博客(3)
- 收藏
- 关注
RSS订阅原创 固定随机数种子 实验复现
固定随机数种子 只需在import 后，主代码前添加： def setup_seed(seed=0): os.environ[‘PYTHONHASHSEED’] = str(seed) torch.manual_seed(seed) torch.cuda.manual_seed(seed) torch.cuda.manual_seed_all(seed) np.random.seed(seed) random.seed(seed) torch.backends.cudnn.deterministic = T
2020-10-12 16:28:24
705
原创 tensorboardX可视化结果
1. install tensorboardX pip安装：pip install tensorboardX conda安装：conda install -c conda-forge tensorboardx 2. tensorboardX的基本使用 # demo.py import numpy as np from tensorboardX import SummaryWriter writer = SummaryWriter(log_dir='scalar') for epoch in range(
2020-06-29 21:02:37
1074
原创 数据处理时，np.ndarray和torch.tensor的纠缠
1. Pytorch 1.1 数据类型 Pytorch的数据预处理torchvision.transforms里面的函数，例如RandomSizedCrop()，全部要求的输入类型为PIL Image，而我们的数据用cv2.imread()读出来的格式时np.ndarray(注意：如果是图片的话，size=(H, W , C), GBR)，所以，需要通过Image.fromarray()，将其转化为PIL Image。(转为Image之前要做GBR2RGB的操作吗？等我查阅。。。) 数据预处理后，直接用to
2020-06-13 11:37:20
1481
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人