# forgery-image-detection

**Repository Path**: HNUDLG/forgery-image-detection

## Basic Information

- **Project Name**: forgery-image-detection
- **Description**: No description available
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 1
- **Created**: 2020-10-18
- **Last Updated**: 2020-12-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# forgery-image-detection

## Training

### Single GPU

```shell script
python train.py --data-dir F:\s2_data\data --arch unet --workers 2 --prefix fl --epoch 10 --batch-size 10 --lr 0.0002 --print-freq 100 --gpu 0
```

### Multiple GPUs

```shell script
python train.py --data-dir F:\s2_data\data --arch unet --workers 2 --prefix fl --epoch 10 --batch-size 10 --lr 0.0002 --print-freq 100 --dist-url tcp://127.0.0.1:23456 --dist-backend nccl --multiprocessing-distributed --world-size 1 --rank 0
```

## Predict

```shell script
python predict.py --data-dir F:\s2_data\data --arch unet --workers 1 --batch-size 1 --print-freq 100 --resume ./weights/unet_ce_25.pt --gpu 0
```