Skip to content

Commit

Permalink
doc: add readme
Browse files Browse the repository at this point in the history
  • Loading branch information
xxxsen committed Jul 4, 2024
1 parent 947c8df commit 9129d7c
Show file tree
Hide file tree
Showing 4 changed files with 47 additions and 2 deletions.
46 changes: 46 additions & 0 deletions README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,46 @@
av-capture
===

由于原先的MovieDataCapture作者把工具闭源了, 只能自己写一个了。

## 使用方式

使用docker进行部署, 对应的`docker-compose.yml`参考下面文件

```yaml
version: "3.1"
services:
av-capture:
image: xxxsen/av-capture:latest
container_name: av-capture
volumes:
- /data/scandir:/scandir
- /data/savedir:/savedir
- /data/datadir:/datadir
- /data/config:/config
command: --config=/config/config.json
```
需要挂载扫描目录(/scandir), 存储目录(/savedir), 数据目录(/datadir)和配置目录(/config), 这几个目录在自己的配置文件中指定。
配置完成后, 使用`docker compose up` 进行刮削, 刮削完成的电影会被存储到`/data/savedir`下。

**NOTE: 程序依赖go-face进行人脸识别, 以用于识别图片中的人脸并进行截图, 这个库需要有对应的模型文件, 可以通过项目目录下的`scripts/download_models.sh`进行下载, 程序会将模型下载到`models`目录, 之后将该目录移动到数据目录(`/datadir`)下即可**

## 基础配置

```json
{
"scan_dir": "/dir/to/scan",
"save_dir": "/dir/to/save/scraped/data",
"data_dir": "/dir/to/save/models/and/cache",
"naming": "naming rule, specify naming rule you want, example: {YEAR}/{NUMBER}"
}
```

|配置项|说明|
|---|---|
|scan_dir|扫描目录, 程序会扫描该目录并对其中的影片进行刮削|
|save_dir|保存目录, 刮削成功的电影会被移动到该目录, 并按`naming`指定的命名规则进行命名|
|data_dir|数据目录, 存储中间文件或者模型文件的|
|naming|命名规则, 可用的命名标签如下:{DATE}, {YEAR}, {MONTH}, {NUMBER}, {ACTOR}|
1 change: 0 additions & 1 deletion config/config.go
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ type Config struct {
ScanDir string `json:"scan_dir"`
SaveDir string `json:"save_dir"`
DataDir string `json:"data_dir"`
ModelDir string `json:"model_dir"`
Naming string `json:"naming"`
PluginConfig map[string]interface{} `json:"plugin_config"`
HandlerConfig map[string]interface{} `json:"handler_config"`
Expand Down
2 changes: 1 addition & 1 deletion main.go
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@ func main() {
if err := translater.Init(); err != nil {
logkit.Fatal("init translater failed", zap.Error(err))
}
if err := image.Init(c.ModelDir); err != nil {
if err := image.Init(filepath.Join(c.DataDir, "models")); err != nil {
logkit.Fatal("init image recognizer failed", zap.Error(err))
}
ss, err := buildSearcher(c.Plugins, c.PluginConfig)
Expand Down
File renamed without changes.

0 comments on commit 9129d7c

Please sign in to comment.