卷积网路CNN分类的模型一般使用包括alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet和Swin transformer等10多种模型
目标检测一般是yolov3、yolov4、yolov5、yolox、PSPnet、faster_rcnn、SDD等
图像分割一般是Unet、mask-rcnn、PSPnet、yolov5-segment等
视频演示找052期:
效果展示图如下:
代码文件展示如下:
运行01.py可以对data文件夹下的图片数据进行训练
运行02.py可以对单张图片进行预测
运行03.py可以生成pyqt的交互界面,通过点击按钮对感兴趣的图片进行加载,然后进行检测识别。