search

puigalex / deteccion-objetos-video

Deteccion de objetos sobre video usando PyTorch
56 stars 49 forks source link

Instrucciones AVX #9

Open Eduardo-az opened 3 years ago

Eduardo-az commented 3 years ago

Hola Alex estoy tratando de realizar el entrenamiento pero al correr "python train.py --model_def config/yolov3-custom.cfg --data_config config/custom.data --pretrained_weights weights/darknet53.conv.74 --batch_size 2" me arroja un mensaje y no me permite realizarlo.

Éste es el mesaje

2020-09-23 22:58:39.177286: F tensorflow/core/platform/cpu_feature_guard.cc:37] The TensorFlow library was compiled to use AVX instructions, but these aren't available on your machine. Abortado (`core' generado)

No sé si tenga que volver a instalar tensorflow. Espero puedas ayudarme, saludos.

puigalex commented 3 years ago

Hola Eduardo,

Pudiste resolver este problema? Cuales son las especificaciones de tu equipo (SO, Procesador, GPU si aplica, version de TF etc)

Eduardo-az commented 3 years ago

Si alex, gracias. Lo que estoy tratando de hacer es correr, en una Movidius Neural Compute Stick, el modelo de yolov3 que entreno después de convertirlo a su Representación Intermedia IR utilizando la librería de Model Optimizer de OpenVINO, pero no he podido utilizar el archivo .xml que obtengo de tal conversión, al correr el ejemplo que trae OpenVINO de detecciópn con Yolov3 obtengo un NaN, no sé si este modelo de yolo es compatible con OpenViNO. Estoy utilizando una sola clase.