Views
No views yet
| Component | Requirement |
|---|---|
| Python | 3.7 or higher |
| PyTorch | 1.7 or higher |
| OpenCV | 4.5+ (headless) |
| Pillow | 8.0+ |
| NumPy | 1.19+ |
1# Clone the repository
2git clone https://github.com/fahimahamed1/NeoNude.git
3cd NeoNude
4
5# Install dependencies
6pip install -r requirements.txtcheckpoints/ directory:checkpoints/
├── cm.lib (~700 MB)
├── mm.lib (~700 MB)
└── mn.lib (~700 MB)1# Default: reads input.png, writes output.png
2python main.py
3
4# Custom input/output paths
5python main.py -i photo.jpg -o result.png
6
7# Help
8python main.py --help| Phase | Type | Description |
|---|---|---|
| 0 | OpenCV | Color correction and normalization |
| 1 | GAN | Clothing mask generation (cm.lib) |
| 2 | OpenCV | Mask refinement |
| 3 | GAN | Anatomical detail detection (mm.lib) |
| 4 | OpenCV | Mask finalization with body annotations |
| 5 | GAN | Final image generation (mn.lib) |
NeoNude/
├── main.py # CLI entry point
├── requirements.txt # Python dependencies
├── README.md
├── src/ # Core package
│ ├── __init__.py # Package metadata
│ ├── config.py # Pipeline configuration (Options)
│ ├── pipeline.py # Pipeline orchestrator (process function)
│ ├── model.py # GAN model, dataset, utilities
│ └── transforms/ # OpenCV image transforms
│ ├── __init__.py
│ ├── annotation.py # Body part data class
│ ├── correct.py # Phase 0: color correction
│ ├── maskref.py # Phase 2: mask refinement
│ └── maskfin.py # Phase 4: mask finalization
└── checkpoints/ # Model weight files (not tracked)Always ensure proper authorization and ethical use before processing any images.
