first-commit

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+third_train.pth filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,13 +1 @@
----
-title: Ocean Faster RCNN
-emoji: 📈
-colorFrom: blue
-colorTo: green
-sdk: gradio
-sdk_version: 3.35.2
-app_file: app.py
-pinned: false
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# ocean-faster-rcnn

model.py

@@ -0,0 +1,14 @@
+import torchvision
+
+def create_fasterrcnn_model(num_classes):
+    model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
+
+    for param in model.parameters():
+        param.requires_grad = False
+
+    # Redefine the ROI head:
+    in_features = model.roi_heads.box_predictor.cls_score.in_features
+    # model.roi_heads.box_predictor = model.roi_heads.box_predictor(in_features, out_features =len(categories))
+    model.roi_heads.box_predictor = torchvision.models.detection.faster_rcnn.FastRCNNPredictor(in_features, num_classes)  # FastRCNNPredictor(in_features, num_classes=len(categories))
+
+    return model

requirements.txt

@@ -0,0 +1,3 @@
+torch==1.12.0
+torchvision==0.13.0
+gradio==3.1.4

third_train.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:663160ca9f7aedfd102ff00c4a36ff3f0115ac8b7a75527c8354067e41f62932
+size 165851031