Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
49 changes: 49 additions & 0 deletions README.es-ES.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@


# ASQuery: Un Modelo Basado en Consultas para la Segmentación de Acciones
## Introducción
Este repositorio proporciona una implementación en PyTorch del artículo "ASQuery: Un Modelo Basado en Consultas para la Segmentación de Acciones".

<div align="center">
<img src="framework.png" width="600px"/>
</div>

## Instalación
Sigue INSTALL.md para instalar las dependencias necesarias.
## Para Replicar Nuestros Resultados en Breakfast
**Preparación del conjunto de datos**
* Descarga el conjunto de datos consultando [MS-TCN](https://github.com/yabufarha/ms-tcn).
* Descomprime el archivo dentro de *./data* (o en otro lugar y crea un enlace a *./data*).
* La estructura de carpetas debería verse así:
```
This folder
│ README.md
│ ...
└───data/
│ └───breakfast/
│ │ └───features/
│ │ └───groundTruth/
│ │ └───splits/
│ │ └───mapping.txt
│ └───...
|
└───libs
│ ...
```
**Entrenamiento y Evaluación**
* Para entrenar nuestro ASQuery. Esto creará una carpeta de experimentos bajo ./ckpt que almacena la configuración de entrenamiento, los registros y los puntos de control.
```shell
python main.py configs/breakfast/config.yaml --output result
```
* [Opcional] Monitorea el entrenamiento usando TensorBoard
```shell
tensorboard --logdir=./ckpt/breakfast/config_result/logs
```
* Evalúa el modelo entrenado.
```shell
python eval.py configs/breakfast/config.yaml ckpt/breakfast/config_result
```
## Agradecimientos
[ActionFormer](https://github.com/happyharrycn/actionformer_release): La base de código principal sobre la que nos basamos, y es un modelo maravilloso.