python小知识

迭代器

[classes[i] for i in [1, 2, 3]]

# 把一个东西变成迭代器，取出其中一个迭代对象
next(iter(dataloader))

引入自己写的包，函数，类

1. 如果要引用的函数文件在同一个目录下，就直接import就行。

2. 如果在不同目录下，要先将该目录加入项目环境变量，如下面的例子。

# 导入自己的包，放在上一个目录文件夹utils里面的utils.py
import sys
sys.path.append("..")
from utils.utils import MyDataset, Net, normalize_invert