Maquina de Vetores de Suporte para Classificacao D
Maquina de Vetores de Suporte para Classificacao D
Resumo: O sistema ferroviário desempenha uma função importante no transporte de carga e de pessoas
de um país. Para garantir a qualidade e segurança dos serviços, as empresas que detêm a concessão de
malhas ferroviárias precisam ter como foco a manutenção e preservação dos trilhos. Este trabalho propõe
uma ferramenta computacional que auxilia na classificação de danos encontrados na superfície dos
boletos dos trilhos a partir de imagens capturadas por câmeras. Os trilhos foram classificados em três
categorias, a depender do grau de severidade do dano apresentado no boleto: trilhos com o boleto em
boas condições, trilhos com o boleto apresentando danos que podem ser reparados com esmerilhamento
e trilhos com boletos apresentando danos severos que exigem substituição imediata. Neste trabalho
foram executadas técnicas de extração de características a partir da análise de textura, espalhamento e
distribuição estatística dos níveis de cinza das imagens dos trilhos. As características extraídas serviram
de dados de entrada para um classificador de Máquina de Vetores de Suporte SVM. O método de
classificação proposto atingiu uma acurácia média de 95,74% e uma precisão média de 96,21%. Os
resultados alcançados mostram que a ferramenta proposta é promissora e pode ser aplicada na inspeção
de trilhos.
Palavras-chave: processamento de imagens; ferrovia; inspeção de ferrovia; Máquina de Vetores de
Suporte.
Abstract: The railway system plays an important role in the logistics of transporting cargo and people
in the country. To guarantee the quality and safety of the railway service, the maintenance of tracks must
always be on the focus of the companies that hold the railway concession. This work proposes a
computational tool that assists classifying damage found on the surfaces of rail billets from images
2 v.9 n.1 2023
captured by cameras. The rails were classified according to three levels of damage severity: rails with
the billet surface in good condition, rails with damage to the billet surface that could still undergo
grinding maintenance, and rails with severe damage to the billet surface and needing immediate
replacement. In this work a method of extracting features from texture, scattering and statistical
distribution of gray levels of pixels were executed and a Support Vector Machine technique was used
as the classifier. The proposed method achieved an average accuracy of 95.74% and an average precision
of 96.21%, proving to be a promising tool for using on railways inspection.
Keywords: image processing; railroad; rail inspection; Support Vector Machine.
problemas superficiais no boleto dos trilhos imagens dos trilhos ferroviários, usando
em oito categorias, alcançando uma uma câmera de alta resolução. Para a etapa
acurácia de 82%. de detecção de objetos foi treinada uma rede
Em Yaman, Karakose, Akin (2017), neural convolucional (CNN) para
os autores extraem curvas características de identificar e localizar diferentes objetos de
cada imagem baseado em suas variâncias ao interesse tais como trilhos, dormentes e
longo de cada linha de pixel da região do fixadores. Na etapa de comparação de
trilho. As entradas do classificador são o imagens, as imagens capturadas são
coeficiente angular, o valor de “a” e o valor comparadas com imagens de referência
de “b”, conforme pode ser visto na Figura 3 usando a técnica de contraste de Weber para
que ilustra as curvas de variâncias. O autor detectar possíveis defeitos nos trilhos. Os
apresenta, nesse trabalho, uma tabela com resultados mostraram que a metodologia
cada tipo de problema do trilho classificado proposta é capaz de detectar com precisão
e com seus parâmetros preestabelecidos. diferentes objetos de interesse nos trilhos
Após a análise das imagens, essa tabela ferroviários, bem como identificar possíveis
serve de base para a classificação do tipo de defeitos, como rachaduras e desgaste
problema encontrado utilizando lógica excessivo. A acurácia média da detecção de
fuzzy. objetos foi de 96,5%, enquanto a acurácia
Em Min et al. (2018) foi utilizada média da detecção de defeitos foi de 92,3%.
visão computacional e técnica de O trabalho apresentado em (GAN et
aprendizado profundo para detecção de al., 2020) propõe um método que compara
defeitos em trilho ferroviários em tempo o fundo de duas imagens e extrai de cada
real. Uma rede neural convolucional (CNN) pixel da imagem características como o
foi treinada para análise de imagens valor do pixel, valor médio, valor da
capturadas da superfície dos trilhos e mediana dos pixels com seus vizinhos. A
identificação de diferentes tipos de defeitos. ideia principal é comparar essas
O pré-processamento envolveu a aplicação características de fundo de um trilho com
de um filtro de mediana para reduzir o ruído defeito com um trilho sem defeito. Uma
nas imagens e, na sequência, a binarização outra característica desse método, é que a
para separar a área de interesse, neste caso avaliação é feita em tempo real através de
o trilho, do restante da imagem. Após esta um computador instalado a bordo do
etapa, foram aplicados filtros morfológicos equipamento que está inspecionando a
para remover ruídos adicionais e destacar os ferrovia.
defeitos. O sistema apresentou uma alta Em Zhai, Ma (2022) foi proposto
precisão na detecção de defeitos em trilhos um algoritmo de detecção de defeitos na
ferroviários, alcançando uma taxa de superfície dos trilhos baseado em um
detecção de 96,7% para defeitos na método de fusão de saliência
superfície do trilho. Outro ponto importante multifuncional. Esse algoritmo combina
apresentado no trabalho é que o sistema é características como textura, cor e
capaz de detectar defeitos em diferentes orientação para identificar áreas de
condições climáticas e em velocidades de interesse e detectar defeitos com maior
até 120 km/h, o que sugere que o sistema precisão. Segundo o autor, a abordagem
pode ser uma ferramenta valiosa para a multifuncional permite que o algoritmo se
manutenção preventiva de trilhos adapte a diferentes condições e tipos de
ferroviários. defeitos, aumentando a eficácia da
A metodologia proposta em Jang et detecção. O algoritmo proposto também
al. (2019) consiste em três etapas principais: emprega uma etapa de pré-processamento
aquisição de imagens, detecção de objetos e para remover o ruído e melhorar a qualidade
comparação de imagens. Na etapa de das imagens dos trilhos. Em seguida, são
aquisição de imagens, são capturadas extraídas as características visuais dos
5 v.9 n.1 2023
"# "#
Energia = √𝐴𝑆𝑀, (2) onde 𝜇0 = ∑'%& 𝑖 × 𝑝0 (𝑖), 𝜇1 = ∑$%& 𝑗 ×
"#
𝑝1 (𝑗) , 𝜎0! = ∑'%&(𝑖 − 𝜇0 )! 𝑝0 (𝑖) e 𝜎1! =
onde p(i,j) é um elemento de uma MC de !
tamanho 𝑁𝑔 × 𝑁𝑔 e representa a ∑"#
$%&X𝑗 − 𝜇1 Y 𝑝1 (𝑗) são as médias e as
probabilidade normalizada de ocorrência variâncias da distribuição marginal
das intensidades de cinza i vizinha ao nível associada a 𝑝(𝑖, 𝑗), com 𝑝0 (𝑖) =
de cinza j, utilizando uma distância 𝑑 = 1 ∑"# "#
$%& 𝑝(𝑖, 𝑗) e 𝑝1 (𝑗) = ∑'%& 𝑝(𝑖, 𝑗).
e ângulo 𝜃 = 0°.
A homogeneidade também mede o
grau de variação local dos níveis de cinza 3.2 CARACTERÍSTICAS DE
entre pares de pixels. Quanto mais DISPERSÃO
constante é uma imagem, maior é esta A variância é uma medida de
medida (SOUTO JUNIOR, 2010). Assim, dispersão que diz o quão distante de uma
com valores mais baixos de média, um valor se encontra (DEVORE,
homogeneidade, aumenta-se a possibilidade 2006). Analisando as imagens dos trilhos do
de haver defeitos no trilho avaliado. O tipo A, tipo B e tipo C, observou-se que
cálculo da homogeneidade a partir da MC é existe um padrão que se repete em todas as
dado pela Eq. 3. situações. Nos trilhos tipo A, sua superfície
é mais homogênea, com pouca variação de
"# "# ((',$)
𝐻𝑜𝑚𝑜𝑔 = ∑'%& ∑$%& &,('-$)! . (3) cor e textura. Já nos trilhos do tipo B e tipo
C, existem deformidades que
descaracterizam a superfície do trilho.
O contraste e a dissimilaridade Desse modo, a variância do nível de cinza
também quantificam a variação local dos
7 v.9 n.1 2023
na imagem do trilho pode ser um parâmetro defeitos ou ranhuras), tipo B (trilho com
que representa o grau de defeito do trilho. falhas ou ranhuras evidentes, mas não
caracterizando defeitos significativos) e
3.3 CARACTERÍSTICAS DE tipo C (trilhos que apresentam defeitos
significativos), conforme mostrado na
DISTRIBUIÇÃO DOS PIXELS
Figura 2.
Além da textura e da dispersão, A técnica de SVM foi a adotada
também foi analisado a tendência de para ser o classificador pois, de acordo com
distribuição dos valores dos pixels nas (BOSER; GUYON; VAPNIK,1992), os
imagens dos boletos dos trilhos. Nessa classificadores treinados com o algoritmo
análise, foram consideradas duas SVM são capazes de obter margens de
características das distribuições: a separação mais amplas e, ao mesmo tempo,
assimetria e a curtose (DEVORE, 2006). apresentam uma baixa taxa de erros de
classificação.
3.3.1 Assimetria Esta técnica busca um limiar de
classificação linear que maximize a
A assimetria, como o nome explica, margem de separação entre as classes das
é uma medida que avalia quão assimétrica é amostras.
uma distribuição de dados. Uma Após o treinamento, o SVM traça o
distribuição é simétrica quando seus valores hiperplano representado pela linha, capaz
de média, mediana e moda coincidem. A de dividir as duas classes de dados. A
comparação entre o valor da média e o valor classificação de novas amostras dá-se pela
da moda, dá, portanto, uma indicação da distância destas à margem de separação
inclinação da distribuição. (hiperplano), descrita por vetores
(MÜLLER; GUIDO, 2016). Neste trabalho
3.3.2 Curtose foi utilizado um SVM com kernel linear por
de ser o mais simples do ponto de vista de
Denomina-se curtose o grau de implementação, e menos propenso à
achatamento da distribuição de frequência ocorrência de overfitting (ROSEBROCK,
de um histograma (DEVORE, 2006). A 2014).
distribuição normal é a distribuição de Outras informações acerca do
referência denominada mesocúrtica; ambiente de desenvolvimento e ferramentas
quando uma distribuição tem uma curva que foram utilizadas neste trabalho são:
mais concentrada (mais aguda em sua parte linguagem de programação Python, por ser
superior), a denominação é leptocúrtica; uma linguagem limpa e concisa com bom
finalmente, quando a distribuição é uma suporte para entradas/saídas numéricas,
curva mais aberta e achatada, é chamada de imagens e plotagens diversas (PILIFRIM,
platicúrtica. No processamento de imagens 2012); o ambiente de desenvolvimento
digitais, os valores de curtose são usado foi o Jupyter (KLUYVER et al.,
interpretados em combinação com ruído e 2016); a biblioteca usada para a criação e
resolução da imagem. Valores altos de manipulação de dados foi a Pandas; para a
curtose podem estar associados com baixo manipulação das imagens e plotagem dos
ruído e baixa resolução (GONZALEZ; gráficos foram usadas a OpenCV
WOODS, 2010). (BRADSKI, 2000), MatplotLib (HUNTER,
2007) e Numpy (VAN DER WALT;
4 CLASSIFICADOR COLBERT; VAROQUAUX, 2011); na
manipulação dos algoritmos de aprendizado
Este trabalho utiliza a técnica de
de máquina foi usada a biblioteca Scikit-
SVM para classificar os trilhos em três tipos
learn. (ROSEBROCK, 2014).
de classes a depender do grau de severidade
dos defeitos dos boletos: tipo A (trilho sem
8 v.9 n.1 2023
95,74% respectivamente, o que mostra que GAN, J. et al. Online Rail Surface
essa é uma ferramenta de uso promissor na Inspection Utilizing Spatial Consistency
operação em campo em um ambiente and Continuity. IEEE Transactions on
parcialmente controlado. Systems, Man, and Cybernetics:
Como trabalhos futuros, novos Systems, v. 50, n. 7, p. 2741–2751, jul.
testes podem ser realizados utilizando 2020.
imagens de outras ferrovias. Além disso,
uma interface amigável deve ser GONZALEZ, R. C.; WOODS, R. E.
desenvolvida para aplicação desse sistema Processamento Digital de Imagens. São
em ambiente industrial. Por fim, propõem o Paulo: Pearson, 2010.
estudo de outras técnicas mais robustas de
aprendizado profundo, que poderiam HARALICK, R. M.; SHANMUGAM, K.;
melhorar o desempenho do sistema DINSTEIN, I. Textural Features for Image
proposto. Classification. IEEE Transactions on
Systems, Man, and Cybernetics, v. SMC-
3, n. 6, p. 610–621, nov. 1973.
AGRADECIMENTOS
Os autores agradecem à Vale pelo
financiamento do projeto “Inspeção HIGA, D. N.; KINA, E. J.; GAY NETO, A.
Automática de Ferrovia e Porto” e pela Wheelset-rail mechanical model for a
cessão das imagens utilizadas neste steady-state dynamic condition and
trabalho. prediction of rolling contact fatigue locci.
Vehicle System Dynamics, v. 60, n. 1, p.
281–308, 2 jan. 2022.
REFERÊNCIAS
ANTT. Extensão da Malha Ferroviária. HUNTER, J. D. Matplotlib: A 2D Graphics
Disponível em: Environment. Computing in Science &
[Link] Engineering, v. 9, n. 3, p. 90–95, 2007.
view/4751/[Link]>. Acesso em: Disponível em:
21 jun. 2019. <[Link]
265/>.
BRINA, H. L. Estradas de ferro. Rio de
Janeiro: LTC, 1983.
JANG, J. et al. Intelligent Image-Based
Railway Inspection System Using Deep
BRADSKI, G. The OpenCV Library. Dr.
Learning-Based Object Detection and
Dobb’s Journal of Software Tools, 2000.
Weber Contrast-Based Image
Comparison. Sensors, v. 19, n. 21, p. 4738,
BOSER, B. E.; GUYON, I. M.; VAPNIK,
31 out. 2019.
V. N. A training algorithm for optimal
margin classifiers. Proceedings of the fifth
KLUYVER, T. et al. Jupyter Notebooks—a
annual workshop on Computational
publishing format for reproducible
learning theory - COLT ’92, 1992.
computational workflows. Positioning and
Power in Academic Publishing: Players,
DEVORE, J. L. Probabilidade e
Agents and Agendas, p. 87–90, 2016.
Estatísticapar a Engenharia e Ciências.
6a. ed. São Paulo: Cengage, 2006.
KOHAVI, R. A study of cross-validation
FACELI, K. et al. Inteligência Artificial: and bootstrap for accuracy estimation and
Uma Abordagem de Aprendizado de model selection. 1995, San Francisco, CA,
Máquina. São Paulo: LTC, 2011. USA: [s.n.], 1995. p. 1137–1143.
12 v.9 n.1 2023
LI, Q.; REN, S. A Visual Detection System ROSEBROCK, A. Practical Phyton and
for Rail Surface Defects. IEEE OpenCV. [S.l: s.n.], 2014
Transactions on Systems, Man, and
Cybernetics, Part C (Applications and SOUTO JUNIOR, C. A. Avaliação de
Reviews), v. 42, n. 6, p. 1531–1542, nov. descritores de textura para segmentação
2012. não-supervisionada de imagens. 2010.
Dissertação (Mestrado) - Faculdade de
PILIFRIM, M. Dive into Python. [S.l: s.n.], Engenharia Elétrica e de Computação,
2012. v. 104. Disponível em: Universidade Estadual de Campinas, 138 f.
<[Link] 2010.
lookup/doi/10.1093/jnci/djs084>.
VAN DER WALT, S.; COLBERT, S. C.;
MA, K. et al. Texture classification for rail VAROQUAUX, G. The NumPy Array: A
surface condition evaluation. mar. 2016, Structure for Efficient Numerical
[S.l.]: IEEE, mar. 2016. p. 1–9. Computation. Computing in Science &
Engineering, v. 13, n. 2, p. 22–30, mar.
MACHADO, A. et al. Manual Técnico da 2011. Disponível em:
Via Permanente. Rev. ed. [S.l.]: VALE S. <[Link]
A., 2009. 236/>.