biganabc / pytorchGUI

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorchGUI

简介

pytorchGUI项目,旨在让不会编程的人也能搭建卷积神经网络,并且进行图像分类任务。

以后会扩展除了卷积神经网络之外的其他神经网络,以及除了图像分类任务之外的其他机器学习任务。

使用

主界面

UUU

创建模型

手工搭建模型

pytorchGUI/2.png at main · biganabc/pytorchGUI (github.com)

创建经典模型

pytorchGUI/3.png at main · biganabc/pytorchGUI (github.com)

训练

pytorchGUI/4.png at main · biganabc/pytorchGUI (github.com)

查看模型结构

pytorchGUI/5.png at main · biganabc/pytorchGUI (github.com)

调用模型

pytorchGUI/6.png at main · biganabc/pytorchGUI (github.com)

环境(需要的Python库)

threading
PyQt5
time
pytorch
pandas
pyqtgraph
os
shutil
PIL
numpy
torchvision
sys
pickle
netron
datetime
inspect

About


Languages

Language:Python 100.0%