文件名称:image_classification
文件大小:16KB
文件格式:ZIP
更新时间:2024-04-28 13:57:12
JupyterNotebook
image_classification 数据集 定义数据集类的实现以进行训练和推理。 动车组 __getitem__方法,将读取train.csv的idx//7行。 每行有7张图像。 通过idx%7决定选择哪个图像。 # 이미지 이름 목록 self . img_name = [ 'normal' , 'mask1' , 'mask2' , 'mask3' , 'mask4' , 'mask5' , 'incorrect_mask' ] # 이미지 포맷 목록 self . img_type_list = [ '.png' , '.jpg' , '.jpeg' ] # csv의 idx//7번째 경로 불러오기 img_path = os . path . join ( self . data_path ,
【文件预览】:
image_classification-main
----AdamP.py(3KB)
----inference.py(1KB)
----train.py(4KB)
----dataset.py(3KB)
----notebook()
--------train()
--------model()
--------dataset()
--------inference()
----model.py(3KB)
----.gitignore(55B)
----README.md(3KB)