Spaces:
Sleeping
Sleeping
File size: 7,694 Bytes
4814c8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 |
# Contributing to ViT Auditing Toolkit
First off, thank you for considering contributing to the ViT Auditing Toolkit! It's people like you that make this tool better for everyone.
## π Ways to Contribute
### 1. Reporting Bugs π
Before creating bug reports, please check existing issues to avoid duplicates. When creating a bug report, include:
- **Clear title and description**
- **Steps to reproduce** the behavior
- **Expected vs actual behavior**
- **Screenshots** if applicable
- **Environment details** (OS, Python version, etc.)
**Example:**
```markdown
**Bug**: GradCAM visualization fails with ViT-Large model
**Steps to reproduce:**
1. Select ViT-Large from dropdown
2. Upload any image
3. Select GradCAM method
4. Click "Analyze Image"
**Expected:** GradCAM heatmap visualization
**Actual:** Error message "AttributeError: ..."
**Environment:**
- OS: Ubuntu 22.04
- Python: 3.10.12
- PyTorch: 2.2.0
```
### 2. Suggesting Features β¨
Feature requests are welcome! Please provide:
- **Clear use case**: Why is this feature needed?
- **Proposed solution**: How should it work?
- **Alternatives considered**: Other approaches you've thought about
- **Additional context**: Screenshots, mockups, references
### 3. Contributing Code π»
#### Development Setup
```bash
# 1. Fork the repository on GitHub
# 2. Clone your fork
git clone https://github.com/YOUR-USERNAME/ViT-XAI-Dashboard.git
cd ViT-XAI-Dashboard
# 3. Create a virtual environment
python -m venv venv
source venv/bin/activate # Windows: venv\Scripts\activate
# 4. Install dependencies
pip install -r requirements.txt
# 5. Install development dependencies
pip install pytest black flake8 mypy
# 6. Create a feature branch
git checkout -b feature/amazing-feature
```
#### Code Style Guidelines
**Python Style:**
- Follow [PEP 8](https://pep8.org/)
- Use 4 spaces for indentation
- Maximum line length: 100 characters
- Use meaningful variable names
**Formatting:**
```bash
# Format code with Black
black src/ tests/ app.py
# Check style with flake8
flake8 src/ tests/ app.py --max-line-length=100
# Type checking with mypy
mypy src/ --ignore-missing-imports
```
**Documentation:**
- Add docstrings to all functions and classes
- Use Google-style docstrings
- Update README.md if adding new features
**Example:**
```python
def explain_attention(model, processor, image, layer_index=6, head_index=0):
"""
Extract and visualize attention weights from a specific layer and head.
Args:
model: Pre-trained ViT model with attention outputs enabled.
processor: Image processor for model input preparation.
image (PIL.Image): Input image to analyze.
layer_index (int): Transformer layer index (0-11 for base model).
head_index (int): Attention head index (0-11 for base model).
Returns:
matplotlib.figure.Figure: Visualization of attention patterns.
Raises:
ValueError: If layer_index or head_index is out of range.
RuntimeError: If attention weights cannot be extracted.
Example:
>>> from PIL import Image
>>> image = Image.open("cat.jpg")
>>> fig = explain_attention(model, processor, image, layer_index=6)
>>> fig.savefig("attention.png")
"""
# Implementation...
```
#### Testing
All new features must include tests:
```bash
# Run all tests
pytest tests/
# Run specific test file
pytest tests/test_explainer.py
# Run with coverage
pytest --cov=src tests/
```
**Writing Tests:**
```python
import pytest
from src.explainer import explain_attention
def test_attention_visualization():
"""Test attention visualization with valid inputs."""
# Setup
model, processor = load_test_model()
image = create_test_image()
# Execute
fig = explain_attention(model, processor, image, layer_index=6)
# Assert
assert fig is not None
assert len(fig.axes) > 0
def test_attention_invalid_layer():
"""Test attention visualization with invalid layer index."""
model, processor = load_test_model()
image = create_test_image()
with pytest.raises(ValueError):
explain_attention(model, processor, image, layer_index=99)
```
#### Commit Messages
Follow the [Conventional Commits](https://www.conventionalcommits.org/) specification:
```
<type>(<scope>): <subject>
<body>
<footer>
```
**Types:**
- `feat`: New feature
- `fix`: Bug fix
- `docs`: Documentation changes
- `style`: Code style changes (formatting, etc.)
- `refactor`: Code refactoring
- `test`: Adding or updating tests
- `chore`: Maintenance tasks
**Examples:**
```
feat(explainer): add LIME explainability method
- Implement LIME-based explanations
- Add visualization function
- Update documentation
Closes #42
```
```
fix(gradcam): resolve tensor dimension mismatch
GradCAM was failing for batch size != 1 due to
incorrect tensor reshaping. Now properly handles
single image inputs.
Fixes #38
```
#### Pull Request Process
1. **Update documentation**: README, docstrings, etc.
2. **Add tests**: Ensure your code is tested
3. **Run tests locally**: All tests must pass
4. **Update CHANGELOG**: Add your changes
5. **Create PR**: Use a clear title and description
**PR Template:**
```markdown
## Description
Brief description of changes
## Type of Change
- [ ] Bug fix
- [ ] New feature
- [ ] Breaking change
- [ ] Documentation update
## Testing
- [ ] All existing tests pass
- [ ] New tests added for new functionality
- [ ] Tested manually with various inputs
## Checklist
- [ ] Code follows style guidelines
- [ ] Documentation updated
- [ ] No new warnings or errors
- [ ] Commit messages are clear
```
### 4. Improving Documentation π
Documentation improvements are always welcome:
- Fix typos or unclear explanations
- Add examples and tutorials
- Improve code comments
- Create video demonstrations
- Translate to other languages
### 5. Reviewing Pull Requests π
Help review open pull requests:
- Test the changes locally
- Provide constructive feedback
- Check for potential issues
- Verify documentation is updated
## π― Good First Issues
Look for issues labeled `good first issue` or `help wanted` - these are great starting points!
## π Project Priorities
Current focus areas:
1. **Stability**: Bug fixes and error handling
2. **Performance**: Optimization for large models
3. **Features**: Additional explainability methods
4. **Documentation**: More examples and tutorials
5. **Testing**: Improved test coverage
## π€ Code of Conduct
### Our Pledge
We are committed to providing a welcoming and inspiring community for all.
### Our Standards
**Positive behavior includes:**
- Being respectful of differing viewpoints
- Gracefully accepting constructive criticism
- Focusing on what is best for the community
- Showing empathy towards other community members
**Unacceptable behavior includes:**
- Harassment, trolling, or discriminatory comments
- Personal or political attacks
- Publishing others' private information
- Other conduct which could reasonably be considered inappropriate
### Enforcement
Instances of abusive, harassing, or otherwise unacceptable behavior may be reported to the project maintainers. All complaints will be reviewed and investigated.
## π¬ Getting Help
- **Questions**: Use [GitHub Discussions](https://github.com/dyra-12/ViT-XAI-Dashboard/discussions)
- **Bugs**: Open an [issue](https://github.com/dyra-12/ViT-XAI-Dashboard/issues)
- **Chat**: Join our community (link coming soon)
## π Thank You!
Your contributions, large or small, make this project better. We appreciate your time and effort!
---
**Happy Contributing! π**
|