HyperAIHyperAI

Command Palette

Search for a command to run...

Détection de vêtements de sécurité et de casques complexes, grands et réalistes : jeu de données et méthode

Fusheng Yu Jiang Li Xiaoping Wang Shaojin Wu Junjie Zhang Zhigang Zeng

Résumé

La détection des vêtements de sécurité et des casques est essentielle pour garantir la sécurité des ouvriers du bâtiment. Toutefois, le développement de modèles d’apprentissage profond dans ce domaine a été freiné par le manque de jeux de données de haute qualité. Dans cette étude, nous construisons un jeu de données volumineux, complexe et réaliste dédié à la détection des vêtements de sécurité et des casques (SFCHD). Ce jeu de données, issu de deux usines chimiques authentiques, comprend 12 373 images, 7 catégories et 50 552 annotations. Nous divisons le jeu de données SFCHD en ensembles d’entraînement et de test selon un ratio de 4:1, et validons son utilité en appliquant plusieurs algorithmes classiques de détection d’objets. Par ailleurs, inspirés par les mécanismes d’attention spatiale et canal, nous proposons un module d’amélioration de l’éclairage faible basé sur l’attention spatiale et canal (SCALE). Ce module est un composant plug-and-play, hautement flexible. Des évaluations étendues du module SCALE sur les jeux de données ExDark et SFCHD démontrent empiriquement son efficacité dans l’amélioration des performances des détecteurs en conditions de faible éclairage. Le jeu de données et le code source sont disponibles publiquement à l’adresse suivante : https://github.com/lijfrank-open/SFCHD-SCALE.


Créer de l'IA avec l'IA

De l'idée au lancement — accélérez votre développement IA avec le co-codage IA gratuit, un environnement prêt à l'emploi et le meilleur prix pour les GPU.

Codage assisté par IA
GPU prêts à l’emploi
Tarifs les plus avantageux

HyperAI Newsletters

Abonnez-vous à nos dernières mises à jour
Nous vous enverrons les dernières mises à jour de la semaine dans votre boîte de réception à neuf heures chaque lundi matin
Propulsé par MailChimp