anodev commited on
Commit
71c8e4c
1 Parent(s): 6b9c6e5

Added first version of the model.

Browse files

Training dataset metadata:
Dataset classes: ['soft', 'digital', 'hard']
Number of classes: 3
Total number of images: 18415
Number of images per class:
soft : 5482
digital : 1206
hard : 11727

The hard class contains the following categories of objects:
object, laptop, charger, pc mouse, pc, rocks, table, bed, box, sneakers, ship, wire, guitar, fork, spoon, plate, keyboard, car, bus, screwdriver, ball, door, flower, clocks, fruit , food, robot.
The soft class contains the following categories of objects:
animal, people, human, man, woman, t-shirt, hairs, hair, dog, cat, monkey, cow, medusa, clothes
The digital class contains the following categories of scenes: screenshot

The classifier uses DenseNet161 as the encoder.

Files changed (1) hide show
  1. scene_classifier.pth +3 -0
scene_classifier.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:856a5ad29ccef0bd0f119f10f049a2755da8ed214576306b12eb76c1b7047d8c
3
+ size 109529839