tiantiandashi的博客

1. 在运行代码的时候在终端设置CUDA_VISIBLE_DEVICES=0 python train.py2.在代码中指定import osos.environ[“CUDA_VISIBLE_DEVICES”] = “0”3. set_deviceimport torchtorch.cuda.set_device(0)以上方式意味着指定使用0号GPU，官方建议使用CUDA_VISIBLE_DEVICES，不建议使用 set_device另外使用gpustat可以查看哪个用户正..

首先先把数据集的图片路径保存在一个txt文件夹里面import osdef generate(dir, label): listText = open('list.txt', 'a') for file in dir: fileType = os.path.split(file) if fileType[1] == '.txt': continue name = file + ' ' + str(int(la

最近要做图片的预处理工作，目的是把原图像的roi提取出来。这里使用python做图像的批量裁剪，代码如下：# coding: utf-8from PIL import Imageimport osimport os.pathimport numpy as npimport cv2imgs = []os_getcwd=os.getcwd().replace('\\','/')...

题目：输入整数数组arr，找出其中最小的k个数。例如，输入4、5、1、6、2、7、3、8这8个数字，则最小的4个数字是1、2、3、4。示例 1：输入：arr = [3,2,1], k = 2输出：[1,2] 或者 [2,1]示例 2：输入：arr = [0,1,2,1], k = 1输出：[0]这道题用python做太简单了。。。class Solu...

题目：给定一个包含大写字母和小写字母的字符串，找到通过这些字母构造成的最长的回文串。在构造过程中，请注意区分大小写。比如"Aa"不能当做一个回文字符串。输入:"abccccdd"输出:7解释:我们可以构造的最长的回文串是"dccaccd", 它的长度是 7。思路：可以观察到回文字符串具有对称性，即从中间切开，左右两边的元素的个数总是偶数对出现。该问题实际上1可以...

天下苦秦久矣。看别人的源码总是被各种下划线带晕了，总算偶得妙文，开疑解惑。原文预估成文于2015之前，作者不详，传播过程中许多格式不幸丢失。本人不忍佳篇失传，斗胆稍加修缮，增补格式，希望不损原意。Python中下划线---完全解读Python 用下划线作为变量前缀和后缀指定特殊变量_xxx 不能用’from module import *’导入_...