BenjiELCA commited on
Commit
4cc36e3
·
1 Parent(s): 144b4fe
Streamlit-Image-Annotation.zip DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:3fac0d3825299fcc1e946a7bcf9ddb23c351f1a7bcfbbf504b2ed447f22ebecb
3
- size 104629368
 
 
 
 
modules/train.py CHANGED
@@ -1,7 +1,5 @@
1
  import copy
2
- import cv2
3
  import numpy as np
4
- import random
5
  import time
6
  import torch
7
  import torchvision.transforms.functional as F
@@ -9,8 +7,6 @@ import matplotlib.pyplot as plt
9
 
10
  from modules.eval import main_evaluation
11
  from torch.optim import SGD, AdamW
12
- from torch.utils.data import DataLoader, Dataset, Subset, ConcatDataset
13
- from torch.utils.data.dataloader import default_collate
14
  from torchvision.models.detection import keypointrcnn_resnet50_fpn, KeypointRCNN_ResNet50_FPN_Weights
15
  from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
16
  from torchvision.models.detection.keypoint_rcnn import KeypointRCNNPredictor
 
1
  import copy
 
2
  import numpy as np
 
3
  import time
4
  import torch
5
  import torchvision.transforms.functional as F
 
7
 
8
  from modules.eval import main_evaluation
9
  from torch.optim import SGD, AdamW
 
 
10
  from torchvision.models.detection import keypointrcnn_resnet50_fpn, KeypointRCNN_ResNet50_FPN_Weights
11
  from torchvision.models.detection.faster_rcnn import FastRCNNPredictor
12
  from torchvision.models.detection.keypoint_rcnn import KeypointRCNNPredictor
modules/utils.py CHANGED
@@ -11,12 +11,6 @@ from torch.utils.data.dataloader import default_collate
11
  import cv2
12
  import matplotlib.pyplot as plt
13
  from torch.utils.data import DataLoader, Subset, ConcatDataset
14
- from tqdm import tqdm
15
- from torch.optim import SGD
16
- import time
17
- from torch.optim import AdamW
18
- import copy
19
- from torchvision import transforms
20
  import streamlit as st
21
 
22
 
 
11
  import cv2
12
  import matplotlib.pyplot as plt
13
  from torch.utils.data import DataLoader, Subset, ConcatDataset
 
 
 
 
 
 
14
  import streamlit as st
15
 
16
 
setup.sh DELETED
@@ -1,7 +0,0 @@
1
- #!/bin/bash
2
-
3
- # Unzip the package
4
- unzip -o Streamlit-Image-Annotation.zip -d Streamlit-Image-Annotation
5
-
6
- # Install the package in editable mode
7
- pip install -e ./Streamlit-Image-Annotation