迭代器
[classes[i] for i in [1, 2, 3]]
# 把一個東西變成迭代器敞掘,取出其中一個迭代對象
next(iter(dataloader))
引入自己寫的包择葡,函數(shù),類
如果要引用的函數(shù)文件在同一個目錄下千诬,就直接import就行娱据。
如果在不同目錄下蚪黑,要先將該目錄加入項目環(huán)境變量,如下面的例子中剩。
# 導入自己的包忌穿,放在上一個目錄文件夾utils里面的utils.py
import sys
sys.path.append("..")
from utils.utils import MyDataset, Net, normalize_invert