- 博客(3)
- 收藏
- 关注
RSS订阅转载 pytorch 指定GPU进行训练
原文链接：https://blog.csdn.net/ax7399/article/details/84557410全局方法：运行的时候调用CUDA_VISIBLE_DEVICES = 1 main.py或者使用os的方法（推荐）：import os#多块使用逗号隔开os.environ['CUDA_VISIBLE_DEVICES'] = '1'非全局方法with torch.cuda.device(1): ......
2020-12-27 15:52:16
369
转载 python3 相对路径 ‘’，‘/’，‘./‘，‘../‘
原文链接：https://blog.csdn.net/Fighting_Yaya/article/details/80275304‘’ ： 当前同级目录‘/’ ：根目录‘./’ ：当前同级目录‘…/’ ：上级目录示例代码如下:f = open("aaa.txt","w") #当前目录f.write("在哪啊")f.close()f = open("/foo.txt","w") #根目录下f.write("Python是一个非常好的语言。\n是的，的确非
2020-12-15 10:21:58
181
原创 显著性目标检测（salient object detection）CRF后处理代码
某位大佬开源了论文源码，github地址：https://github.com/Xiaoqi-Zhao-DLUT/GateNet-RGB-Saliency里面有一部分CRF后处理预测图的代码，这里我就直接扒过来了。使用方法：if args['crf_refine']: prediction = crf_refine(np.array(img), np.array(prediction))源码：import pydensecrf.densecrf as dcrfdef crf_refine(
2020-12-15 10:11:35
954
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人