- 博客(4)
- 收藏
- 关注
原创 基于mobilenetv2做一个人脸属性多标签分类模型
因项目需求需要对人像进行男、女、老、少人脸属性分类,与单标签多类不同(一张图一个标签,有多个类别),多标签分类(一个图有多个标签如:男的老人、女的青年人等)的标注应该如何实现?参考,采用onehot标签编码实现数据的多标签标注。
2024-05-07 17:40:06
607
2
原创 从零开始复现AFFormer(语义分割)训练自己的数据集并转onnx——linux
1)作者训练网络为:AFFormer_tiny_cityscapes。修改configs/AFFormer/AFFormer_tiny_cityscapes.py文件,把第一个箭头的路径改为自己下载的官方提供的预训练权重的路径,把第二个箭头改为自己的类别数(作者按人体皮肤部位与背景分为了两类)。2)修改configs/_base_/datasets/cityscapes_1024x1024.py与configs/_base_/datasets/cityscapes.py。修改为自己的类别,颜色随便填。
2023-10-16 14:41:59
1525
17
原创 从零开始复现seaformer(语义分割)训练自己的数据集——linux
2)在seaformer-seg下创建modelzoos文件夹,在modelzoos文件夹下创建classification文件夹用于存放seaformer提供的预训练权重文件,将下载的预训练权重改名为SeaFormer_S.pth。修改SeaFormer-main\seaformer-seg\local_configs\seaformer\seaformer_small.py,改为自己的类别数并将第一行中的SyncBN改为BN。记录一下,若存在不对的地方,请大家多多包涵~3)安装mmcv-full。
2023-10-09 16:32:11
2996
33
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人