使用DSFD检测DarkFace数据集过程

1.下载Dark Face数据集,使用track2.2_test_sample文件中图片进行人脸检测测试。

2.修改DSFD源码中demo.py部分:

test_oneimage():
def test_oneimage():
    # load net

    # 影响网络的自动求导机制,使网络前向传播后不进行求导和反向传播(仅测试时使用)
    torch.set_grad_enabled(False)

    # 加载config配置参数
    cfg = widerface_640
    # 分类的类别数目---widerface.py
    num_classes = len(WIDERFace_CLASSES) + 1 # +1 background
    # 加载SSD网络模型,返回一个SSD实例
    net = build_ssd('test', cfg['min_dim'], num_classes) # initialize SSD
    # 加载预训练模型train_model
    net.load_state_dict(torch.load(args.trained_model))
    net.cuda()
    # 表示进入评估模式,神经网络中有train(),eval()两种模式,使用eval()可关闭dropout
    net.eval()
    print('Finished loading model!')

    # evaluation
    cuda = args.cuda
    transform = TestBaseTransform((104, 117, 123))
    thresh=cfg['conf_thresh']
    #save_path = args.save_folder
  
  • 2
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
1. OpenCVSharp - https://github.com/shimat/opencvsharp 2. EmguCV - https://github.com/emgucv/emgucv 3. FaceNetDotNet - https://github.com/sooftware/FaceNetDotNet 4. InsightFace - https://github.com/deepinsight/insightface 5. FaceRecognitionDotNet - https://github.com/ageitgey/face_recognition 6. DlibDotNet - https://github.com/takuya-takeuchi/DlibDotNet 7. MTCNN-Tensorflow - https://github.com/AITTSMD/MTCNN-Tensorflow 8. RetinaFace - https://github.com/deepinsight/insightface/tree/master/detection/retinaface 9. InsightFace_Pytorch - https://github.com/TreB1eN/InsightFace_Pytorch 10. FaceDetection-DSFD - https://github.com/Tencent/FaceDetection-DSFD 11. FaceDetection-DSFD-Pytorch - https://github.com/hukkelas/FaceDetection-DSFD-Pytorch 12. FaceRecognition - https://github.com/ageitgey/face_recognition 13. Face-Recognition-Tensorflow - https://github.com/krasserm/face-recognition-tensorflow 14. Face-Recognition - https://github.com/YuvalNirkin/face_recognition 15. Face-Recognition-Using-Deep-Learning - https://github.com/abhijeet3922/Face-Recognition-Using-Deep-Learning 16. DeepFaceLab - https://github.com/iperov/DeepFaceLab 17. DeepID - https://github.com/deepinsight/insightface/tree/master/recognition/arcface_torch 18. Face-Recognition-Using-Deep-Learning - https://github.com/abhijeet3922/Face-Recognition-Using-Deep-Learning 19. Face-Recognition-Attendance-System - https://github.com/shubham101096/Face-Recognition-Attendance-System 20. Face-Recognition-System - https://github.com/akshaybahadur21/Face-Recognition-System 以上是20个ml.net框架下第三方开源项目,关于人脸识别的,大家可以根据自己的需求进行选择。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值