reusable_code
### jupyter notebook 中执行代码指定GPU
import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"
### resize image
def resize_image(ori_dir, des_dir,classes):
for cc ...
原创
2020-01-09 11:22:12 ·
169 阅读 ·
0 评论