torch Dataset类
生活随笔
收集整理的這篇文章主要介紹了
torch Dataset类
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
實現(xiàn)Dataset類:
-
__init__:初始化信息
-
__getitem__:傳入的圖片路徑和下標(biāo)返回目錄下的圖片信息
-
__len__:返回圖片的張數(shù)
總結(jié)
以上是生活随笔為你收集整理的torch Dataset类的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 安装labelImg对数据集标注
- 下一篇: Tensorflow tf.keras.