V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
OilMoe
V2EX  ›  程序员

求一个少量图片就能实现物体检测项目

  •  
  •   OilMoe · 2023-09-06 22:27:37 +08:00 · 1369 次点击
    这是一个创建于 428 天前的主题,其中的信息可能已经有所发展或是发生改变。

    公司最近要折腾这种东西,想要实现的功能类似拍照打卡的感觉。但是我也不是这个方向的,能做的就是上 yolo 系列打打标签,训练了一下。但效果不尽人意,总是在一些比较奇怪的角度下,将某些不在数据集中的东西识别成了其他物体。不仅数据集数量也很少(普遍十几张,多点也就 60 张左右),而且物体有大有小,唯一优势可能就是物体本身不会发生形变。公司也有 4090 的显卡,不太担心显存的问题。

    5 条回复    2023-09-07 11:03:08 +08:00
    dayeye2006199
        1
    dayeye2006199  
       2023-09-06 23:14:59 +08:00   ❤️ 1
    60 几张其实也差不多了,记得做一下图片增量 -- 旋转,翻转,调调亮度,尽量让模型见一下一些奇怪的角度
    RVRGnXxwawP5kwKW
        2
    RVRGnXxwawP5kwKW  
       2023-09-07 00:41:41 +08:00 via iPhone   ❤️ 1
    可以在某个大型相似任务的数据集上训练,然后在自己的数据集上 finetune 。
    zzzzzzy
        3
    zzzzzzy  
       2023-09-07 09:57:12 +08:00   ❤️ 1
    一楼的思路是正确的,拍照打卡这种需求,每台手机拍出来的照片也大不相同,各个手机品牌对于拍照的算法也不一样,摄像头也不一样,你需要根据少量的数据集的基础上拓展这个数据集,包括一楼提到的旋转、翻转、调整亮度、锐化、放大、缩小等操作,这样才能提供识别成功率。
    OilMoe
        4
    OilMoe  
    OP
       2023-09-07 10:56:11 +08:00
    @dayeye2006199 #1 yolo 5 自带了数据增强的配置文件,使用它的可以吗?还有请问一下,是否可以将沿路拍的东西(不是选择的标志物,比如地板,墙壁)放在训练集中,来防止模型将地板和标志物混淆这种情况有帮助吗
    OilMoe
        5
    OilMoe  
    OP
       2023-09-07 11:03:08 +08:00
    @zzzzzzy #3 yolo 5 是否可以将沿路拍的东西(不是选择的标志物,比如地板,墙壁)放在训练集中,来防止模型将地板和标志物混淆这种情况有帮助吗
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1003 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 23ms · UTC 21:32 · PVG 05:32 · LAX 13:32 · JFK 16:32
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.