hongyaohongyao/smart_classroom_demo

一些问题想请教一下

coisino opened this issue · 2 comments

我有几个问题:

  1. 要是改成CPU的要怎么改呀
  2. 为什么要把yolov5转成torchscript呢
  3. 这个项目的目标检测使用的是yolov5,该怎么把它改成别的模型呢,比如其他的yolo版本
  4. 我试着看了一下里边多线程相关的代码,没太看懂。。。请问有相关的教程嘛
  1. yolov5用官方代码转的有点毛病,所以项目整体目前只支持gpu,如果要使用cpu需要用yolov5的官方项目导出cpu版的yolov5s.torchscript.pt权重。代码里面把smart_classroom文件夹里面几个xxx_app.py文件的device变量从‘cuda’改成‘cpu’
  2. 转成torchscript后权重就不依赖使用python代码定义的类,方便之后在其他平台部署,事实上我部分功能也做了java的版本,可以用java的opencv演示,不过因为时间和其他一些原因,最后还是用pyqt做的demo
  3. 改成其他版本只要导出对应的torchscript模型权重,然后修改smart_classroom文件夹里面几个xxx_app.py文件的yolov5_weight变量就行
  4. 多线程相关的代码分两个部分,pyqt的多线程你可以系统学一下pyqt,一般会教,因为有些功能不用多线程实现不了。如果是说pipline_module部分的多线程,这个就是自己写的一个流水线处理框架,算是致敬AlphaPose,事实上python因为gil的缘故做流水线处理也提高不了速度(不过在java里试的是有用😜)

还有一个问题,动作识别要是添加新的动作的话,要修改一下网络结构然后重新训练,请问有相关的训练的代码嘛