Update train_net.py
Browse filesCo-Authored-By: Nasheed Yasin <nasheed.ny@gmail.com>
- tools/train_net.py +1 -1
tools/train_net.py
CHANGED
@@ -26,7 +26,7 @@ import pandas as pd
|
|
26 |
def get_augs(cfg):
|
27 |
"""Add all the desired augmentations here. A list of availble augmentations
|
28 |
can be found here:
|
29 |
-
|
30 |
"""
|
31 |
augs = [
|
32 |
T.ResizeShortestEdge(
|
|
|
26 |
def get_augs(cfg):
|
27 |
"""Add all the desired augmentations here. A list of availble augmentations
|
28 |
can be found here:
|
29 |
+
https://detectron2.readthedocs.io/en/latest/modules/data_transforms.html
|
30 |
"""
|
31 |
augs = [
|
32 |
T.ResizeShortestEdge(
|