NanjingForestryUniversity/cotton_color
3
1
Fork 0
mirror of https://github.com/Karllzy/cotton_color.git synced 2025-11-08 18:53:53 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

修改代码

Browse Source
This commit is contained in:
wrz1-zzzzz 2024-11-15 10:22:41 +08:00
parent ebd1f6924a
commit 73c952202f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
DPL/dgd_class/predict.py
View File

@ -8,7 +8,7 @@ from torchvision import datasets, models, transforms
import os
import matplotlib.pyplot as plt
from test import device, class_names, imshow
from train import device, class_names, imshow
# 加载已训练的 ONNX 模型