AutoWebCompat - Automatically detect web compatibility issues
The aim of this project is creating a tool to automatically detect web compatibility issues without human intervention.
The project uses Selenium to collect web page screenshots automatically on Firefox and Chrome.
The crawler loads web pages from the URLs on the webcompat.com tracker and tries to reproduce the reported issues by interacting with the elements of the page. As soon as the page is loaded and after every interaction with the elements, the crawler takes a screenshot.
The crawler repeats the same steps in Firefox and Chrome, generating a set of comparable screenshots.
data/ directory contains the screenshots generated by the crawler (N.B.: This directory is not present in the repository itself, but it will be created automatically after you setup the project as described in the Setup paragraph).
Now that we have a dataset with labels, we can train a neural network to automatically detect screenshots that are incompatible. We are currently using a Siamese architecture with different Convolutional Neural Networks, but are open to test other ideas.
We plan to employ three training methodologies:
- Training from scratch on the entire training set;
- Finetuning a network previously pretrained on ImageNet (or other datasets);
- Finetuning a network previously pretrained in an unsupervised fashion.
For the unsupervised training, we are using a related problem for which we already have labels (detecting screenshots belonging to the same website). The pre-training can be helpful because we have plenty of data (as we don't need to manually label them) and we can fine-tune the network we pre-train for our problem of interest.
Structure of the project
- The autowebcompat/utils.py module contains some utility functions;
- The autowebcompat/network.py module contains neural network definition, optimizers definition, along with the loss and accuracy;
- The collect.py script is the crawler that collects screenshots of web pages in different browsers;
- The label.py script is a utility that helps labelling couples of screenshots (are they the same in the two browsers or are there differences?);
- The pretrain.py script trains a neural network on the website screenshots for a slightly different problem (for which we know the solution), so that we can reuse the network weights for the training on the actual problem;
- The train.py script trains the neural network on the website screenshots to detect compat issues;
- The data_inconsistencies.py script checks the generated screenshots and takes note of any data inconsistency (e.g. screenshots that were taken in Firefox but not in Chrome).
Python 3 is required.
- Install Git Large File Storage, either manually or through a package like
git-lfsif available on your system (in case of using PackageCloud).
- Clone the repository with submodules:
git lfs clone --recurse-submodules REPO_URL
- Install the dependencies in requirements.txt:
pip install -r requirements.txt.
- Install the dependencies in test-requirements.txt:
pip install -r test-requirements.txt.
Training the network
The pretrain.py or train.py script can be run to train the neural network, with the following options:
-network To select which network architecture to use -optimizer To select the optimizer to use -classification_type Either Y vs N + D or Y + N vs D --early_stoppping (Optional) To stop training when validation accuracy has stopped improving
Real-time communication for this project happens on Mozilla's IRC network, irc.mozilla.org, in the #webcompat channel.